Spaces:
Runtime error
Runtime error
made minor changed for readability
Browse files- mixtral_agent.py +2 -7
mixtral_agent.py
CHANGED
@@ -33,7 +33,6 @@ llm = ChatOllama(
|
|
33 |
model="mistral:instruct",
|
34 |
base_url= OLLMA_BASE_URL
|
35 |
)
|
36 |
-
prompt = ChatPromptTemplate.from_template("Tell me a short joke about {topic}")
|
37 |
|
38 |
|
39 |
tools = [
|
@@ -71,11 +70,8 @@ agent_executor = AgentExecutor(
|
|
71 |
)
|
72 |
|
73 |
|
74 |
-
|
75 |
if __name__ == "__main__":
|
76 |
|
77 |
-
# global variable for collecting sources
|
78 |
-
|
79 |
input = agent_executor.invoke(
|
80 |
{
|
81 |
"input": "How to generate videos from images using state of the art macchine learning models; Using the axriv retriever " +
|
@@ -94,7 +90,7 @@ if __name__ == "__main__":
|
|
94 |
# }
|
95 |
# )
|
96 |
|
97 |
-
#
|
98 |
# {
|
99 |
# "input": "I am looking for a text to 3d model; Using the google search tool " +
|
100 |
# "add the urls in the final answer using the metadata from the retriever, also provid a summary of the searches"
|
@@ -102,5 +98,4 @@ if __name__ == "__main__":
|
|
102 |
# }
|
103 |
# )
|
104 |
|
105 |
-
x = 0
|
106 |
-
|
|
|
33 |
model="mistral:instruct",
|
34 |
base_url= OLLMA_BASE_URL
|
35 |
)
|
|
|
36 |
|
37 |
|
38 |
tools = [
|
|
|
70 |
)
|
71 |
|
72 |
|
|
|
73 |
if __name__ == "__main__":
|
74 |
|
|
|
|
|
75 |
input = agent_executor.invoke(
|
76 |
{
|
77 |
"input": "How to generate videos from images using state of the art macchine learning models; Using the axriv retriever " +
|
|
|
90 |
# }
|
91 |
# )
|
92 |
|
93 |
+
# input_2 = agent_executor.invoke(
|
94 |
# {
|
95 |
# "input": "I am looking for a text to 3d model; Using the google search tool " +
|
96 |
# "add the urls in the final answer using the metadata from the retriever, also provid a summary of the searches"
|
|
|
98 |
# }
|
99 |
# )
|
100 |
|
101 |
+
x = 0 # for debugging purposes
|
|