Skip to content

Commit 41fc4d2

Browse files
committed
fix retrieval agent config error
1 parent dbe34ac commit 41fc4d2

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

Diff for: litemultiagent/agents/base.py

-1
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,6 @@ def send_prompt(self, content: str) -> str:
100100
self.messages.append({"role": "user", "content": content})
101101
return self._send_completion_request()
102102

103-
# ... (rest of the BaseAgent methods remain the same)
104103
def _send_completion_request(self, depth: int = 0) -> str:
105104
if depth >= 8:
106105
return None

Diff for: litemultiagent/main.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,8 @@ def main():
124124
file_retrieval_agent_config,
125125
db_retrieval_agent_config
126126
],
127-
"agent_description": "Use a database retrieval agent to fetch information based on a given query.",
128-
"parameter_description": "The query to be processed by the database retrieval agent."
127+
"agent_description": "Use a smart research assistant to look up information using multiple sources including web search, database retrieval, and local file retrieval.",
128+
"parameter_description": "The task description specifying the information source (web search, database, local file) and the question to be answered. specify this in natural language"
129129
}
130130
# retrieval_agent = agent_manager.get_agent(retrieval_agent_config)
131131
# # # Example usage
@@ -152,13 +152,13 @@ def main():
152152
main_agent = agent_manager.get_agent(main_agent_config)
153153

154154
# # # Example usage
155-
task = "generate a image of a ginger cat and save it as ginger_cat.png"
156-
result = main_agent.execute(task)
157-
print("IO Agent Result:", result)
158-
159-
task = "write python script to calculate the sum from 1 to 10, and run the python script to get result"
160-
result = main_agent.execute(task)
161-
print("IO Agent Result:", result)
155+
# task = "generate a image of a ginger cat and save it as ginger_cat.png"
156+
# result = main_agent.execute(task)
157+
# print("IO Agent Result:", result)
158+
#
159+
# task = "write python script to calculate the sum from 1 to 10, and run the python script to get result"
160+
# result = main_agent.execute(task)
161+
# print("IO Agent Result:", result)
162162

163163
task = "browse web to search and check the brands of dining table, and summarize the results in a table, save the table into a markdown file called summary.md"
164164
result = main_agent.execute(task)

0 commit comments

Comments
 (0)