Research agent directory in examples

ShoggothAI · Apr 26, 2024 · bb691e8 · bb691e8
1 parent 43cd69c
commit bb691e8
Show file tree

Hide file tree

Showing 4 changed files with 293 additions and 302 deletions.
diff --git a/...crew/agent/research/question_generator.py → ...ples/research_agent/question_generator.py b/...crew/agent/research/question_generator.py → ...ples/research_agent/question_generator.py
diff --git a/motleycrew/tool/llm_tool.py b/motleycrew/tool/llm_tool.py
@@ -18,27 +18,28 @@ def __init__(
         description: str,
         prompt: str | BasePromptTemplate,
         llm: Optional[BaseLanguageModel] = None,
+        input_description: Optional[str] = "Input for the tool.",
     ):
-        langchain_tool = create_llm_langchain_tool(name, description, prompt, llm)
+        langchain_tool = create_llm_langchain_tool(
+            name=name, description=description, prompt=prompt, llm=llm, input_description=input_description
+        )
         super().__init__(langchain_tool)
 
 
 def create_llm_langchain_tool(
     name: str,
     description: str,
     prompt: str | BasePromptTemplate,
-    llm: Optional[BaseLanguageModel] = None,
-    input_description: Optional[str] = "Input for the tool.",
+    llm: Optional[BaseLanguageModel],
+    input_description: Optional[str],
 ):
     if llm is None:
         llm = init_llm(llm_framework=LLMFramework.LANGCHAIN)
 
     if not isinstance(prompt, BasePromptTemplate):
         prompt = PromptTemplate.from_template(prompt)
 
-    assert (
-        len(prompt.input_variables) == 1
-    ), "Prompt must contain exactly one input variable"
+    assert len(prompt.input_variables) == 1, "Prompt must contain exactly one input variable"
     input_var = prompt.input_variables[0]
 
     class LLMToolInput(BaseModel):