Update multiagents.py

multiagents.py

@@ -55,14 +55,34 @@ dotenv.load_dotenv()
 #     max_tokens=4000,
 # )
 
-
-#InferenceClientModel InferenceClient
-My_Agent = InferenceClientModel(
-    provider="groq",
-    api_key=os.getenv("GROQ_API_KEY"),  # Your Replicate API key
-    model_id = "qwen/qwen3-32b"
-)
-
+# SETUP AND TEST
+
+try:
+    #InferenceClientModel InferenceClient
+    My_Agent = InferenceClientModel(
+        provider="groq",
+        api_key=os.getenv("GROQ_API_KEY"),
+        model_id = "qwen/qwen3-32b"
+    )
+
+    output_test = MyAgent('What are the 3 laws of robotics')
+    print('1', output_test)
+
+except:
+    # Check if API key is available
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    if not groq_api_key:
+        raise ValueError("GROQ_API_KEY environment variable is not set")
+    
+    # Fixed model initialization - use LiteLLMModel instead of InferenceClientModel for Groq
+    My_Agent = LiteLLMModel(
+        model_id="groq/qwen3-32b",  # Use a valid Groq model ID
+        api_key=groq_api_key,
+        temperature=0.1,
+        max_tokens=4000,
+    )
+    output_test = MyAgent('What are the 3 laws of robotics')
+    print('2', output_test)
 
 
 # My_Agent = client.chat.completions.create(