LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 16:43:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve tool call handling and UI adjustments

Browse Source
This commit is contained in:
Siddhant Rai
2025-02-13 05:02:10 +05:30
parent e209699b19
commit 8a3612e56c
4 changed files with 61 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
application/tools/agent.py
View File

@@ -144,7 +144,11 @@ class Agent:
if isinstance(resp, str):
yield resp
return
if hasattr(resp, "message") and hasattr(resp.message, "content"):
if (
hasattr(resp, "message")
and hasattr(resp.message, "content")
and resp.message.content is not None
):
yield resp.message.content
return
@@ -152,7 +156,11 @@ class Agent:
if isinstance(resp, str):
yield resp
elif hasattr(resp, "message") and hasattr(resp.message, "content"):
elif (
hasattr(resp, "message")
and hasattr(resp.message, "content")
and resp.message.content is not None
):
yield resp.message.content
else:
completion = self.llm.gen_stream(