Spaces:
Running
Running
pattonma
commited on
Commit
•
9ec925a
1
Parent(s):
b2f993e
getting agents to pass muster
Browse files- helper_functions.py +2 -2
helper_functions.py
CHANGED
|
@@ -17,7 +17,7 @@ def process_file(file):
|
|
| 17 |
|
| 18 |
documents = []
|
| 19 |
if file.path.endswith(".pdf"):
|
| 20 |
-
loader =
|
| 21 |
docs = loader.load()
|
| 22 |
documents.extend(docs)
|
| 23 |
else:
|
|
@@ -80,7 +80,7 @@ def create_team_agent(llm, tools, system_prompt, agent_name, team_members):
|
|
| 80 |
llm,
|
| 81 |
tools,
|
| 82 |
f"{system_prompt}\nBelow are files currently in your directory:\n{{current_files}}",
|
| 83 |
-
team_members
|
| 84 |
)
|
| 85 |
|
| 86 |
def create_agent_node(agent, name):
|
|
|
|
| 17 |
|
| 18 |
documents = []
|
| 19 |
if file.path.endswith(".pdf"):
|
| 20 |
+
loader = PyMuPDFLoader(temp_file)
|
| 21 |
docs = loader.load()
|
| 22 |
documents.extend(docs)
|
| 23 |
else:
|
|
|
|
| 80 |
llm,
|
| 81 |
tools,
|
| 82 |
f"{system_prompt}\nBelow are files currently in your directory:\n{{current_files}}",
|
| 83 |
+
#team_members
|
| 84 |
)
|
| 85 |
|
| 86 |
def create_agent_node(agent, name):
|