Update agents/use_case_agent.py
Browse files- agents/use_case_agent.py +1 -1
agents/use_case_agent.py
CHANGED
|
@@ -4,7 +4,7 @@ from langchain.llms import OpenAI
|
|
| 4 |
from langchain.chains import LLMChain
|
| 5 |
|
| 6 |
def get_use_case_agent(llm):
|
| 7 |
-
prompt_template = PromptTemplate.from_file("prompts/use_case_prompt.txt")
|
| 8 |
chain = LLMChain(llm=llm, prompt=prompt_template)
|
| 9 |
|
| 10 |
return Tool(
|
|
|
|
| 4 |
from langchain.chains import LLMChain
|
| 5 |
|
| 6 |
def get_use_case_agent(llm):
|
| 7 |
+
prompt_template = PromptTemplate.from_file("prompts/use_case_prompt.txt", input_variables=["input"])
|
| 8 |
chain = LLMChain(llm=llm, prompt=prompt_template)
|
| 9 |
|
| 10 |
return Tool(
|