Fix

agent.py

@@ -296,16 +296,37 @@ def build_graph(provider: str = "huggingface"):
     
     if provider == "huggingface":
         # Use a more capable model from HuggingFace
-        endpoint = HuggingFaceEndpoint(
-            repo_id="microsoft/DialoGPT-large",  # You can also try "google/flan-t5-xl" or "bigscience/bloom-7b1"
-            temperature=0.1,
-            huggingfacehub_api_token=hf_token,
-            model_kwargs={
-                "max_length": 1024,
-                "return_full_text": False
-            }
-        )
-        llm = ChatHuggingFace(llm=endpoint)
+        try:
+            # Try with a well-supported model first
+            endpoint = HuggingFaceEndpoint(
+                repo_id="google/flan-t5-base",  # This model works well with the current setup
+                temperature=0.1,
+                huggingfacehub_api_token=hf_token,
+                max_new_tokens=512,
+                task="text2text-generation"
+            )
+            llm = ChatHuggingFace(llm=endpoint)
+        except Exception as e:
+            print(f"Failed to initialize google/flan-t5-base: {e}")
+            # Fallback to another model
+            try:
+                endpoint = HuggingFaceEndpoint(
+                    repo_id="microsoft/DialoGPT-medium",
+                    temperature=0.1,
+                    huggingfacehub_api_token=hf_token,
+                    max_new_tokens=512
+                )
+                llm = ChatHuggingFace(llm=endpoint)
+            except Exception as e2:
+                print(f"Failed to initialize DialoGPT-medium: {e2}")
+                # Final fallback
+                endpoint = HuggingFaceEndpoint(
+                    repo_id="bigscience/bloom-560m",
+                    temperature=0.1,
+                    huggingfacehub_api_token=hf_token,
+                    max_new_tokens=256
+                )
+                llm = ChatHuggingFace(llm=endpoint)
     else:
         raise ValueError("Only 'huggingface' provider is supported in this version.")