codeembed 0.1.0__tar.gz → 0.1.1__tar.gz

{codeembed-0.1.0 → codeembed-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeembed
-Version: 0.1.0
+Version: 0.1.1
 Summary: Embeds your codebase and makes it available for quick LLM lookups via MCP.
 Project-URL: Homepage, https://github.com/robino16/codeembed
 Project-URL: Repository, https://github.com/robino16/codeembed

{codeembed-0.1.0 → codeembed-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codeembed"
-version = "0.1.0"
+version = "0.1.1"
 description = "Embeds your codebase and makes it available for quick LLM lookups via MCP."
 readme = "README.md"
 requires-python = ">=3.11"

{codeembed-0.1.0 → codeembed-0.1.1}/src/codeembed/llm/openai_adapter.py

@@ -1,13 +1,15 @@
-from typing import List, Optional, Type, TypeVar, cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, List, Optional, Type, TypeVar, cast
 
-from openai import OpenAI
-from openai._types import omit
-from openai.types.chat import ChatCompletionMessageParam
 from pydantic import BaseModel
 
 from codeembed.llm.base import LLMServiceBase
 from codeembed.llm.models import ChatMessage, LLMResponse, StructuredLLMResponse
 
+if TYPE_CHECKING:
+    from openai import OpenAI
+
 T = TypeVar("T", bound=BaseModel)
 
 
@@ -23,6 +25,8 @@ class OpenAILLMService(LLMServiceBase):
         max_tokens: Optional[int] = None,
         temperature: Optional[float] = None,
     ) -> StructuredLLMResponse[T]:
+        from openai._types import omit
+        from openai.types.chat import ChatCompletionMessageParam
 
         openai_messages = cast(List[ChatCompletionMessageParam], messages)
 
@@ -61,6 +65,8 @@ class OpenAILLMService(LLMServiceBase):
         max_tokens: Optional[int] = None,
         temperature: Optional[float] = None,
     ) -> LLMResponse:
+        from openai._types import omit
+        from openai.types.chat import ChatCompletionMessageParam
 
         openai_messages = cast(List[ChatCompletionMessageParam], messages)