Handle API key for ollama servers (#1257)

This was missed.

Signed-off-by: Juan Antonio Osorio <[email protected]>

Author: JAORMX

src/codegate/types/ollama/_generators.py

@@ -55,13 +55,18 @@ async def generate_streaming(request, api_key, base_url):
 
 
 async def streaming(request, api_key, url, cls):
+    headers = dict()
+
+    if api_key:
+        headers["Authorization"] = f"Bearer {api_key}"
     payload = request.json(exclude_defaults=True)
     if os.getenv("CODEGATE_DEBUG_OLLAMA") is not None:
         print(payload)
 
     client = httpx.AsyncClient()
     async with client.stream(
             "POST", url,
+            headers=headers,
             content=payload,
             timeout=300, # TODO this should not be hardcoded
     ) as resp: