Pass instructions and requires_tools to /models

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.6.0.dev17"
+version = "0.6.0.dev18"
 description = ""
 authors = ["Guillaume Aquilina <guillaume@workflowai.com>"]
 readme = "README.md"

workflowai/core/client/_models.py

@@ -202,6 +202,11 @@ class ListModelsResponse(Page[ModelInfo]):
     """Response from the list models API endpoint."""
 
 
+class ListModelsRequest(BaseModel):
+    instructions: Optional[str] = Field(default=None, description="Used to detect internal tools")
+    requires_tools: Optional[bool] = Field(default=None, description="Whether the agent uses external tools")
+
+
 class CompletionsResponse(BaseModel):
     """Response from the completions API endpoint."""
 

workflowai/core/client/agent.py

@@ -12,6 +12,7 @@
     CompletionsResponse,
     CreateAgentRequest,
     CreateAgentResponse,
+    ListModelsRequest,
     ListModelsResponse,
     ModelInfo,
     ReplyRequest,
@@ -477,7 +478,11 @@ def _sanitize_validator(cls, kwargs: RunParams[AgentOutput], default: OutputVali
         validator = kwargs.pop("validator", default)
         return validator, cast(BaseRunParams, kwargs)
 
-    async def list_models(self) -> list[ModelInfo]:
+    async def list_models(
+        self,
+        instructions: Optional[str] = None,
+        requires_tools: Optional[bool] = None,
+    ) -> list[ModelInfo]:
         """Fetch the list of available models from the API for this agent.
 
         Returns:
@@ -486,12 +491,22 @@ async def list_models(self) -> list[ModelInfo]:
         Raises:
             ValueError: If the agent has not been registered (schema_id is None).
         """
+
         if not self.schema_id:
             self.schema_id = await self.register()
 
-        response = await self.api.get(
+        request_data = ListModelsRequest(instructions=instructions, requires_tools=requires_tools)
+
+        if instructions is None and self.version and isinstance(self.version, VersionProperties):
+            request_data.instructions = self.version.instructions
+
+        if requires_tools is None and self._tools:
+            request_data.requires_tools = True
+
+        response = await self.api.post(
             # The "_" refers to the currently authenticated tenant's namespace
             f"/v1/_/agents/{self.agent_id}/schemas/{self.schema_id}/models",
+            data=request_data,
             returns=ListModelsResponse,
         )
         return response.items