kodit 0.3.14__tar.gz → 0.3.15__tar.gz

{kodit-0.3.14 → kodit-0.3.15}/Makefile

@@ -1,7 +1,10 @@
 # Makefile for Kodit
 
 # Generate OpenAPI json schema from the FastAPI app
-openapi:
+build:
+	uv build
+
+openapi: build
 	uv run src/kodit/utils/dump_openapi.py --out docs/reference/api/ kodit.app:app
 
 openapi-check: openapi

{kodit-0.3.14 → kodit-0.3.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kodit
-Version: 0.3.14
+Version: 0.3.15
 Summary: Code indexing for better AI code generation
 Project-URL: Homepage, https://docs.helixml.tech/kodit/
 Project-URL: Documentation, https://docs.helixml.tech/kodit/

{kodit-0.3.14 → kodit-0.3.15}/docs/reference/api/index.md

@@ -12,7 +12,7 @@ look at the [hosted version](https://kodit.helix.ml/docs).
 This is the REST API for the Kodit server. Please refer to the
 [Kodit documentation](https://docs.helix.ml/kodit/) for more information.
     
-Current version: 0.3.14
+Current version: 0.3.15
 
 ## Authentication
 

{kodit-0.3.14 → kodit-0.3.15}/docs/reference/api/openapi.json

@@ -3,7 +3,7 @@
   "info": {
     "title": "kodit API",
     "description": "\nThis is the REST API for the Kodit server. Please refer to the\n[Kodit documentation](https://docs.helix.ml/kodit/) for more information.\n    ",
-    "version": "0.3.14"
+    "version": "0.3.15"
   },
   "paths": {
     "/healthz": {

{kodit-0.3.14 → kodit-0.3.15}/src/kodit/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.14'
-__version_tuple__ = version_tuple = (0, 3, 14)
+__version__ = version = '0.3.15'
+__version_tuple__ = version_tuple = (0, 3, 15)

{kodit-0.3.14 → kodit-0.3.15}/src/kodit/infrastructure/embedding/embedding_providers/openai_embedding_provider.py

@@ -52,7 +52,17 @@ class OpenAIEmbeddingProvider(EmbeddingProvider):
     def _get_encoding(self) -> "Encoding":
         """Return (and cache) the tiktoken encoding for the chosen model."""
         if self._encoding is None:
-            self._encoding = tiktoken.encoding_for_model(self.model_name)
+            try:
+                self._encoding = tiktoken.encoding_for_model(self.model_name)
+            except KeyError:
+                # If the model is not supported by tiktoken, use a default encoding
+                self.log.info(
+                    "Model not supported by tiktoken, using default encoding",
+                    model_name=self.model_name,
+                    default_encoding="o200k_base",
+                )
+                self._encoding = tiktoken.get_encoding("o200k_base")
+
         return self._encoding
 
     def _split_sub_batches(

kodit-0.3.14/tests/kodit/infrastructure/embedding/embedding_provider/test_openai_embedding_provider.py → kodit-0.3.15/tests/kodit/infrastructure/embedding/embedding_provider/openai_embedding_provider_test.py

@@ -260,3 +260,24 @@ class TestOpenAIEmbeddingProvider:
 
         # Should handle malformed response gracefully by returning empty results
         assert len(results) == 0
+
+    @pytest.mark.asyncio
+    async def test_non_openai_model_name(self) -> None:
+        """Test embedding with a non-OpenAI model name."""
+        mock_client = MagicMock()
+        mock_response = MagicMock()
+        mock_response.data = [MagicMock(embedding=[0.1] * 1500)]
+        mock_client.embeddings.create = AsyncMock(return_value=mock_response)
+
+        provider = OpenAIEmbeddingProvider(
+            openai_client=mock_client, model_name="non-openai-model"
+        )
+
+        # This should not crash
+        test_requests = [EmbeddingRequest(snippet_id=1, text="test")]
+        await anext(provider.embed(test_requests))
+
+        # Verify the custom model was used
+        mock_client.embeddings.create.assert_called_once_with(
+            input=["test"], model="non-openai-model"
+        )

kodit-0.3.14/tests/kodit/mcp_test.py

@@ -1,88 +0,0 @@
-"""Tests for the MCP server implementation."""
-
-import pytest
-from fastmcp import Client
-from mcp.types import TextContent
-from sqlalchemy.ext.asyncio import AsyncSession
-
-from kodit.domain.value_objects import FileProcessingStatus
-from kodit.infrastructure.sqlalchemy.entities import (
-    File,
-    Index,
-    Snippet,
-    Source,
-    SourceType,
-)
-from kodit.mcp import mcp
-
-
-@pytest.mark.asyncio
-async def test_mcp_server_basic_functionality(session: AsyncSession) -> None:
-    """Test basic MCP server functionality with real database."""
-    # Create test data
-    source = Source(
-        uri="file:///test/repo",
-        cloned_path="/tmp/test/repo",  # noqa: S108
-        source_type=SourceType.FOLDER,
-    )
-    session.add(source)
-    await session.flush()
-
-    index = Index(source_id=source.id)
-    session.add(index)
-    await session.flush()
-
-    file = File(
-        created_at=source.created_at,
-        updated_at=source.updated_at,
-        source_id=source.id,
-        mime_type="text/plain",
-        uri="file:///test/repo/example.py",
-        cloned_path="/tmp/test/repo/example.py",  # noqa: S108
-        sha256="abc123",
-        size_bytes=100,
-        extension="py",
-        file_processing_status=FileProcessingStatus.CLEAN,
-    )
-    session.add(file)
-    await session.flush()
-
-    snippet = Snippet(
-        file_id=file.id,
-        index_id=index.id,
-        content="def hello_world():\n    return 'Hello, World!'",
-        summary="Simple hello world function",
-    )
-    session.add(snippet)
-    await session.commit()
-
-    # Test MCP client connection
-    async with Client(mcp) as client:
-        # Test tool listing
-        tools = await client.list_tools()
-        assert len(tools) == 2
-        tool_names = {tool.name for tool in tools}
-        assert "search" in tool_names
-        assert "get_version" in tool_names
-
-        # Test version tool
-        result = await client.call_tool("get_version")
-        assert len(result.content) == 1
-        content = result.content[0]
-        assert isinstance(content, TextContent)
-        assert content.text is not None
-
-        # Test search tool
-        result = await client.call_tool(
-            "search",
-            {
-                "user_intent": "Find hello world functions",
-                "related_file_paths": [],
-                "related_file_contents": [],
-                "keywords": ["hello", "world"],
-            },
-        )
-        assert len(result.content) == 1
-        content = result.content[0]
-        assert isinstance(content, TextContent)
-        assert content.text is not None