ai-microcore 4.2.2__tar.gz → 4.2.3__tar.gz

{ai_microcore-4.2.2 → ai_microcore-4.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-microcore
-Version: 4.2.2
+Version: 4.2.3
 Summary: # Minimalistic Foundation for AI Applications
 Keywords: llm,large language models,ai,similarity search,ai search,gpt,openai,framework,adapter
 Author-email: Vitalii Stepanenko <mail@vitaliy.in>

{ai_microcore-4.2.2 → ai_microcore-4.2.3}/microcore/__init__.py

@@ -186,4 +186,4 @@ __all__ = [
     # "wrappers",
 ]
 
-__version__ = "4.2.2"
+__version__ = "4.2.3"

{ai_microcore-4.2.2 → ai_microcore-4.2.3}/microcore/mcp.py

@@ -25,12 +25,24 @@ class WrongMcpUsage(BadAIAnswer):
 
 
 class HeaderAuth(httpx.Auth):
-    def __init__(self, header_name, header_value):
-        self.header_name = header_name
-        self.header_value = header_value
+    def __init__(self, header_name_or_dict: str | dict, header_value: str | None = None):
+        if isinstance(header_name_or_dict, dict):
+            if header_value is not None:
+                raise ValueError(
+                    "HeaderAuth: If a dictionary is provided in first argument, "
+                    "header_value should not be set."
+                )
+            self.headers = header_name_or_dict
+        else:
+            if header_value is None:
+                raise ValueError(
+                    "HeaderAuth: header_value must be provided when header_name is a string"
+                )
+            self.headers = {str(header_name_or_dict): str(header_value)}
 
     def auth_flow(self, request):
-        request.headers[self.header_name] = self.header_value
+        for k, v in self.headers.items():
+            request.headers[k] = v
         yield request
 
 
@@ -293,12 +305,8 @@ class MCPServer:
             self.name = MCPServer.name_from_url(self.url)
         if not self.transport:
             self.transport = self._guess_transport_type_by_url(self.url)
-        if self.auth and isinstance(self.auth, dict) and len(self.auth) == 1:
-            header_name, header_value = next(iter(self.auth.items()))
-            self.auth = HeaderAuth(
-                header_name=header_name,
-                header_value=header_value
-            )
+        if self.auth and isinstance(self.auth, dict):
+            self.auth = HeaderAuth(self.auth)
 
     async def connect(
         self,