isayahc commited on
Commit
793cb2b
1 Parent(s): 5c9cba1

made minor changed for readability

Browse files
Files changed (1) hide show
  1. 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
- # input_1 = agent_executor.invoke(
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