agenthub-python 0.3.2__tar.gz → 0.3.3__tar.gz

{agenthub_python-0.3.2 → agenthub_python-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agenthub-python
-Version: 0.3.2
+Version: 0.3.3
 Summary: AgentHub is the LLM API Hub for the Agent era, built for high-precision autonomous agents.
 Keywords: agent,llm,gemini,claude,gpt
 Author: PrismShadow

{agenthub_python-0.3.2 → agenthub_python-0.3.3}/agenthub/auto_client.py

@@ -77,6 +77,10 @@ class AutoLLMClient(LLMClient):
             from .deepseek_v4 import DeepSeekV4Client
 
             return DeepSeekV4Client(model=model, api_key=api_key, base_url=base_url)
+        elif "openai" in client_type and "embedding" in client_type:
+            from .openai_embedding import OpenaiEmbeddingClient
+
+            return OpenaiEmbeddingClient(model=model, api_key=api_key, base_url=base_url)
         elif "openai" in client_type:
             from .openai import OpenaiClient
 
@@ -84,7 +88,7 @@ class AutoLLMClient(LLMClient):
         else:
             raise ValueError(
                 f"{client_type} is not supported. "
-                "Supported client types: gemini-3, claude-4-8, claude-4-7, claude-4-6, gpt-5.5, gpt-5.4, glm-5.1, kimi-k2.6, kimi-k2.5, deepseek-v4, openai."
+                "Supported client types: gemini-3, claude-4-8, claude-4-7, claude-4-6, gpt-5.5, gpt-5.4, glm-5.1, kimi-k2.6, kimi-k2.5, deepseek-v4, openai-embedding, openai."
             )
 
     def transform_uni_config_to_model_config(self, config: UniConfig) -> Any:

agenthub_python-0.3.3/agenthub/openai_embedding/__init__.py

@@ -0,0 +1,18 @@
+# Copyright 2025 Prism Shadow. and/or its affiliates
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .client import OpenaiEmbeddingClient
+
+
+__all__ = ["OpenaiEmbeddingClient"]

agenthub_python-0.3.3/agenthub/openai_embedding/client.py

@@ -0,0 +1,80 @@
+# Copyright 2025 Prism Shadow. and/or its affiliates
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+from typing import Any, AsyncIterator
+
+from openai import AsyncOpenAI
+
+from ..base_client import LLMClient
+from ..types import UniConfig, UniEvent, UniMessage
+
+
+class OpenaiEmbeddingClient(LLMClient):
+    """OpenAI Embeddings-compatible client implementation."""
+
+    def __init__(self, model: str, api_key: str | None = None, base_url: str | None = None):
+        """Initialize OpenAI-compatible embedding client with model, API key, and base URL."""
+        self._model = model
+        api_key = api_key or os.getenv("OPENAI_API_KEY")
+        base_url = base_url or os.getenv("OPENAI_BASE_URL")
+        self._client = AsyncOpenAI(api_key=api_key, base_url=base_url)
+        self._history: list[UniMessage] = []
+
+    def transform_uni_config_to_model_config(self, config: UniConfig) -> dict[str, Any]:
+        """Transform universal configuration to OpenAI Embeddings configuration."""
+        params: dict[str, Any] = {"model": self._model}
+        embedding_config = config.get("embedding_config") or {}
+        if embedding_config.get("dimensions") is not None:
+            params["dimensions"] = embedding_config["dimensions"]
+        return params
+
+    def transform_uni_message_to_model_input(self, messages: list[UniMessage]) -> list[str]:
+        """Transform universal messages to OpenAI Embeddings input strings."""
+        texts = []
+        for msg in messages:
+            msg_text = ""
+            for item in msg["content_items"]:
+                if item["type"] != "text":
+                    raise ValueError("OpenAI embeddings only support text content items.")
+                msg_text += item["text"]
+            texts.append(msg_text or " ")
+        return texts
+
+    def transform_model_output_to_uni_event(self, model_output: Any) -> UniEvent:
+        """Transform OpenAI Embeddings response to universal event format."""
+        usage = getattr(model_output, "usage", None)
+        return {
+            "role": "assistant",
+            "event_type": "stop",
+            "content_items": [{"type": "embedding", "embedding": item.embedding} for item in model_output.data],
+            "usage_metadata": {
+                "cached_tokens": None,
+                "prompt_tokens": usage.prompt_tokens if usage else None,
+                "thoughts_tokens": None,
+                "response_tokens": None,
+            },
+            "finish_reason": "stop",
+        }
+
+    async def _streaming_response_internal(
+        self,
+        messages: list[UniMessage],
+        config: UniConfig,
+    ) -> AsyncIterator[UniEvent]:
+        """Generate embeddings using OpenAI Embeddings-compatible API."""
+        params = self.transform_uni_config_to_model_config(config)
+        params["input"] = self.transform_uni_message_to_model_input(messages)
+        result = await self._client.embeddings.create(**params)
+        yield self.transform_model_output_to_uni_event(result)

{agenthub_python-0.3.2 → agenthub_python-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agenthub-python"
-version = "0.3.2"
+version = "0.3.3"
 description = "AgentHub is the LLM API Hub for the Agent era, built for high-precision autonomous agents."
 keywords = ["agent", "llm", "gemini", "claude", "gpt"]
 readme = "README.md"