github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-10 13:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Changes review spec ui message

Browse Source
This commit is contained in:
matija-ilijas
2024-09-09 15:21:18 +02:00
parent 98d31986a6
commit 26de2e5cb8
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/agents/spec_writer.py
View File

@@ -201,10 +201,11 @@ class SpecWriter(BaseAgent):
async def review_spec(self, desc: str, spec: str) -> str:
convo = AgentConvo(self).template("review_spec", desc=desc, spec=spec)
llm = self.get_llm(SPEC_WRITER_AGENT_NAME, stream_output=True)
await self.send_message("\n\nAdditional info/examples:\n\n")
llm = self.get_llm(SPEC_WRITER_AGENT_NAME)
llm_response: str = await llm(convo, temperature=0)
additional_info = llm_response.strip()
if additional_info and len(additional_info) > 6:
spec += "\n\nAdditional info/examples:\n\n" + additional_info
await self.send_message(f"\n\nAdditional info/examples:\n\n {additional_info}")
return spec