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

Update retrieving + agent pipeline (#71)

Browse Source
This commit is contained in:
Tuan Anh Nguyen Dang (Tadashi_Cin)
2023-11-14 16:40:13 +07:00
committed by GitHub
parent 693ed39de4
commit 640962e916
8 changed files with 65 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,7 +4,7 @@ from ....base import BaseComponent
from ....llms import PromptTemplate
from ..base import BaseLLM, BaseTool
from ..output.base import BaseScratchPad
from .prompt import zero_shot_planner_prompt
from .prompt import few_shot_planner_prompt, zero_shot_planner_prompt
class Planner(BaseComponent):
@@ -56,7 +56,7 @@ class Planner(BaseComponent):
)
else:
if self.examples is not None:
return zero_shot_planner_prompt.populate(
return few_shot_planner_prompt.populate(
tool_description=worker_desctription,
fewshot=fewshot,
task=instruction,