Update agent.py

agent.py

@@ -45,18 +45,18 @@ from llama_index.core.agent.workflow import AgentWorkflow
 
 # Initialize model (defaults to llama3.1)
 def init_model(model:str='llama3.1:8b-instruct-q4_0'):
-    from llama_index.llms.ollama import Ollama
-    # from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
-
-    llm_model = Ollama(
-        model=model,
-        request_timeout=360.0
-        )
-
-    # llm_model = HuggingFaceInferenceAPI(
-    #     model='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/'
-    #     # model_name='Qwen/Qwen2.5-Coder-32B-Instruct'
-    # )
+    # from llama_index.llms.ollama import Ollama
+    from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
+
+    # llm_model = Ollama(
+    #     model=model,
+    #     request_timeout=360.0
+    #     )
+
+    llm_model = HuggingFaceInferenceAPI(
+        model='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/'
+        # model_name='Qwen/Qwen2.5-Coder-32B-Instruct'
+    )
 
     return llm_model