Initial commit

skills/pocketflow/scripts/test_llm_connection.py

@@ -0,0 +1,76 @@
+#!/usr/bin/env python3
+"""
+Quick script to test your LLM connection
+"""
+
+import os
+import sys
+
+def test_openai():
+    """Test OpenAI connection"""
+    try:
+        from openai import OpenAI
+        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[{"role": "user", "content": "Say 'hello'"}]
+        )
+        print("✅ OpenAI: Connected")
+        print(f"   Response: {response.choices[0].message.content}")
+        return True
+    except Exception as e:
+        print(f"❌ OpenAI: Failed - {e}")
+        return False
+
+def test_anthropic():
+    """Test Anthropic connection"""
+    try:
+        from anthropic import Anthropic
+        client = Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
+        response = client.messages.create(
+            model="claude-3-5-haiku-20241022",
+            max_tokens=100,
+            messages=[{"role": "user", "content": "Say 'hello'"}]
+        )
+        print("✅ Anthropic: Connected")
+        print(f"   Response: {response.content[0].text}")
+        return True
+    except Exception as e:
+        print(f"❌ Anthropic: Failed - {e}")
+        return False
+
+def test_google():
+    """Test Google Gemini connection"""
+    try:
+        from google import genai
+        client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        response = client.models.generate_content(
+            model='gemini-2.0-flash-exp',
+            contents="Say 'hello'"
+        )
+        print("✅ Google Gemini: Connected")
+        print(f"   Response: {response.text}")
+        return True
+    except Exception as e:
+        print(f"❌ Google Gemini: Failed - {e}")
+        return False
+
+if __name__ == "__main__":
+    print("🔍 Testing LLM connections...\n")
+
+    results = {
+        "OpenAI": test_openai(),
+        "Anthropic": test_anthropic(),
+        "Google": test_google()
+    }
+
+    print("\n" + "="*50)
+    working = [k for k, v in results.items() if v]
+    if working:
+        print(f"✅ Working providers: {', '.join(working)}")
+    else:
+        print("❌ No working providers found")
+        print("\nMake sure you've set environment variables:")
+        print("  export OPENAI_API_KEY=sk-...")
+        print("  export ANTHROPIC_API_KEY=sk-ant-...")
+        print("  export GEMINI_API_KEY=...")