Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-03-16 20:50:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

response.tool_calls() methods

Browse source 
Refs https://github.com/simonw/llm/issues/936#issuecomment-2869082672
This commit is contained in:
Simon Willison 2025-05-10 11:38:51 -07:00
parent 7bc2f78156
commit 5aac6b76f0
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
llm/models.py
View file

@ -393,7 +393,7 @@ class _BaseResponse:
self._key = key
self._chunks: List[str] = []
self._done = False
self.tool_calls: List[ToolCall] = []
self._tool_calls: List[ToolCall] = []
self.response_json = None
self.conversation = conversation
self.attachments: List[Attachment] = []
@ -412,7 +412,7 @@ class _BaseResponse:
raise ValueError(f"{self.model} does not support tools")
def add_tool_call(self, tool_call: ToolCall):
self.tool_calls.append(tool_call)
self._tool_calls.append(tool_call)
def set_usage(
self,
@ -617,6 +617,10 @@ class Response(_BaseResponse):
def text_or_raise(self) -> str:
return self.text()
def tool_calls(self) -> List[ToolCall]:
self._force()
return self._tool_calls
def json(self) -> Optional[Dict[str, Any]]:
self._force()
return self.response_json
@ -763,6 +767,10 @@ class AsyncResponse(_BaseResponse):
await self._force()
return "".join(self._chunks)
async def tool_calls(self) -> List[ToolCall]:
await self._force()
return self._tool_calls
async def json(self) -> Optional[Dict[str, Any]]:
await self._force()
return self.response_json