diff --git a/context.go b/context.go
index bb5620a..8320269 100644
--- a/context.go
+++ b/context.go
@@ -47,6 +47,14 @@ func (c *Context) Request() Request {
 	return c.request
 }
 
+func (c *Context) Response() *ResponseChoice {
+	return c.response
+}
+
+func (c *Context) ToolCall() *ToolCall {
+	return c.toolcall
+}
+
 func (c *Context) WithContext(ctx context.Context) *Context {
 	return &Context{Context: ctx, request: c.request, response: c.response, toolcall: c.toolcall}
 }