修改quick think response category 临时if逻辑

This commit is contained in:
Yizhou Chi 2024-08-12 15:32:55 +08:00
parent c8cc67cdb2
commit e0cc3e8d1b
2 changed files with 2 additions and 6 deletions

View file

@ -129,7 +129,7 @@ For requests that are unclear, lack sufficient detail, or are outside the system
- Out of Expertise: Requests for specialized advice (e.g., medical or legal advice) or highly technical tasks beyond the model's scope.
- Unrealistic Scope: Overly broad requests that are impossible to address meaningfully in a single response (e.g., "Tell me everything about...").
**Note:** Before categorizing a request as TASK, consider whether the user has provided sufficient information to proceed with the task. If the request is complex but lacks essential details, it may fall under AMBIGUOUS.
**Note:** Before categorizing a request as TASK, consider whether the user has provided sufficient information to proceed with the task. If the request is complex but lacks essential details or the mentioned files, it should fall under AMBIGUOUS.
{examples}

View file

@ -256,17 +256,13 @@ class RoleZero(Role):
context = self.llm.format_msg(memory + [UserMessage(content=QUICK_THINK_PROMPT)])
intent_result = await self.llm.aask(context)
if "QUICK" in intent_result:
# llm call with the original context
if "QUICK" in intent_result or "AMBIGUOUS " in intent_result: # llm call with the original context
async with ThoughtReporter(enable_llm_stream=True) as reporter:
await reporter.async_report({"type": "quick"})
answer = await self.llm.aask(self.llm.format_msg(memory))
elif "SEARCH" in intent_result:
query = "\n".join(str(msg) for msg in memory)
answer = await SearchEnhancedQA().run(query)
elif "AMBIGUOUS " in intent_result:
# TODO: out of domain, ask human for help
pass
if answer:
self.rc.memory.add(AIMessage(content=answer, cause_by=RunCommand))