workspace working

test_ollama_auth_with_key.py

@@ -0,0 +1,36 @@
+import asyncio
+import aiohttp
+import json
+
+async def test_embed_with_key(key):
+    url = "http://localhost:11434/api/embed"
+    payload = {
+        "model": "snowflake-arctic-embed:latest",
+        "input": ["test"]
+    }
+    headers = {"Content-Type": "application/json"}
+    if key:
+        headers["Authorization"] = f"Bearer {key}"
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.post(url, json=payload, headers=headers) as resp:
+                print(f"Status: {resp.status}")
+                if resp.status == 200:
+                    data = await resp.json()
+                    print("Success")
+                else:
+                    text = await resp.text()
+                    print("Error:", text)
+        except Exception as e:
+            print(f"Exception: {e}")
+
+if __name__ == "__main__":
+    # test with empty key
+    print("Testing with empty key:")
+    asyncio.run(test_embed_with_key(""))
+    # test with dummy key
+    print("\nTesting with dummy key 'governor':")
+    asyncio.run(test_embed_with_key("governor"))
+    # test with no Authorization header
+    print("\nTesting with no Authorization header:")
+    asyncio.run(test_embed_with_key(None))