Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[AUR-395, AUR-415] Adopt Example1 Injury pipeline; add .flow() for enabling bottom-up pipeline execution (#32)

Browse Source 
* add example1/injury pipeline example
* add dotenv
* update various api
This commit is contained in:
ian_Cin
2023-10-02 16:24:56 +07:00
committed by GitHub
parent 3cceec63ef
commit d83c22aa4e
16 changed files with 114 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
knowledgehub/pipelines/agents/react/agent.py
View File

@@ -162,7 +162,7 @@ class ReactAgent(BaseAgent):
prompt = self._compose_prompt(instruction)
logging.info(f"Prompt: {prompt}")
response = self.llm(prompt, stop=["Observation:"]) # type: ignore
response_text = response.text[0]
response_text = response.text
logging.info(f"Response: {response_text}")
action_step = self._parse_output(response_text)
if action_step is None:

4
knowledgehub/pipelines/agents/rewoo/agent.py
View File

@@ -245,7 +245,7 @@ class RewooAgent(BaseAgent):
# Plan
planner_output = planner(instruction)
plannner_text_output = planner_output.text[0]
plannner_text_output = planner_output.text
plan_to_es, plans = self._parse_plan_map(plannner_text_output)
planner_evidences, evidence_level = self._parse_planner_evidences(
plannner_text_output
@@ -263,7 +263,7 @@ class RewooAgent(BaseAgent):
# Solve
solver_output = solver(instruction, worker_log)
solver_output_text = solver_output.text[0]
solver_output_text = solver_output.text
return AgentOutput(
output=solver_output_text, cost=total_cost, token_usage=total_token