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: missing yield in tool agent

Browse Source
This commit is contained in:
Siddhant Rai
2024-12-19 17:55:58 +05:30
parent daa332aa20
commit f67b79f007
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
application/tools/agent.py
View File

@@ -120,7 +120,9 @@ class Agent:
def gen(self, messages):
# Generate initial response from the LLM
if self.llm.supports_tools():
self._simple_tool_agent(messages)
resp = self._simple_tool_agent(messages)
for line in resp:
yield line
else:
resp = self.llm.gen_stream(model=self.gpt_model, messages=messages)
for line in resp: