feat: Add basic support for integration tools to ToolStep

agents-api/agents_api/autogen/Common.py

@@ -9,6 +9,16 @@
 from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, RootModel
 
 
+class JinjaTemplate(RootModel[str]):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    root: str
+    """
+    A valid jinja template.
+    """
+
+
 class Limit(RootModel[int]):
     model_config = ConfigDict(
         populate_by_name=True,

agents-api/agents_api/autogen/Sessions.py

@@ -43,6 +43,16 @@ class CreateSessionRequest(BaseModel):
     """
     Action to start on context window overflow
     """
+    forward_tool_results: Literal["auto"] | StrictBool = "auto"
+    """
+    Whether to forward the tool results to the model when available.
+    "true" => always forward
+    "false" => never forward
+    "auto" => forward if applicable
+
+    If a tool call is made, the tool's output will be sent back to the model as the model's input.
+    If a tool call is not made, the model's output will be returned as is.
+    """
     metadata: dict[str, Any] | None = None
 
 
@@ -70,6 +80,16 @@ class PatchSessionRequest(BaseModel):
     """
     Action to start on context window overflow
     """
+    forward_tool_results: Literal["auto"] | StrictBool = "auto"
+    """
+    Whether to forward the tool results to the model when available.
+    "true" => always forward
+    "false" => never forward
+    "auto" => forward if applicable
+
+    If a tool call is made, the tool's output will be sent back to the model as the model's input.
+    If a tool call is not made, the model's output will be returned as is.
+    """
     metadata: dict[str, Any] | None = None
 
 
@@ -97,6 +117,16 @@ class Session(BaseModel):
     """
     Action to start on context window overflow
     """
+    forward_tool_results: Literal["auto"] | StrictBool = "auto"
+    """
+    Whether to forward the tool results to the model when available.
+    "true" => always forward
+    "false" => never forward
+    "auto" => forward if applicable
+
+    If a tool call is made, the tool's output will be sent back to the model as the model's input.
+    If a tool call is not made, the model's output will be returned as is.
+    """
     id: Annotated[UUID, Field(json_schema_extra={"readOnly": True})]
     metadata: dict[str, Any] | None = None
     created_at: Annotated[AwareDatetime, Field(json_schema_extra={"readOnly": True})]
@@ -160,6 +190,16 @@ class UpdateSessionRequest(BaseModel):
     """
     Action to start on context window overflow
     """
+    forward_tool_results: Literal["auto"] | StrictBool = "auto"
+    """
+    Whether to forward the tool results to the model when available.
+    "true" => always forward
+    "false" => never forward
+    "auto" => forward if applicable
+
+    If a tool call is made, the tool's output will be sent back to the model as the model's input.
+    If a tool call is not made, the model's output will be returned as is.
+    """
     metadata: dict[str, Any] | None = None
 
 
@@ -194,6 +234,16 @@ class CreateOrUpdateSessionRequest(CreateSessionRequest):
     """
     Action to start on context window overflow
     """
+    forward_tool_results: Literal["auto"] | StrictBool = "auto"
+    """
+    Whether to forward the tool results to the model when available.
+    "true" => always forward
+    "false" => never forward
+    "auto" => forward if applicable
+
+    If a tool call is made, the tool's output will be sent back to the model as the model's input.
+    If a tool call is not made, the model's output will be returned as is.
+    """
     metadata: dict[str, Any] | None = None