refactor: renamed models.py to data_models.py and updated the dependencies

deepakdgupta1 · deepakdgupta1 · commit 91cb3475906e · 2025-12-20T18:47:13.000+05:30
diff --git a/scripts/evaluate.py b/scripts/evaluate.py
@@ -7,7 +7,7 @@
 from knowcode.storage.vector_store import VectorStore
 from knowcode.retrieval.hybrid_index import HybridIndex
 from knowcode.llm.embedding import OpenAIEmbeddingProvider
-from knowcode.models import EmbeddingConfig, CodeChunk
+from knowcode.data_models import EmbeddingConfig, CodeChunk
 
 def evaluate(ground_truth_path: Path, index_path: Path) -> dict:
     """Evaluate retrieval quality against ground truth."""
diff --git a/src/knowcode/__init__.py b/src/knowcode/__init__.py
@@ -2,5 +2,5 @@
 
 __version__ = "0.1.0"
 
-from knowcode.models import CodeChunk, EmbeddingConfig
+from knowcode.data_models import CodeChunk, EmbeddingConfig
 from knowcode.storage.chunk_repository import ChunkRepository, InMemoryChunkRepository
diff --git a/src/knowcode/analysis/context_synthesizer.py b/src/knowcode/analysis/context_synthesizer.py
@@ -4,7 +4,7 @@
 from typing import Optional
 
 from knowcode.storage.knowledge_store import KnowledgeStore
-from knowcode.models import Entity, EntityKind
+from knowcode.data_models import Entity, EntityKind
 from knowcode.utils.token_counter import TokenCounter
 
 
diff --git a/src/knowcode/analysis/signals.py b/src/knowcode/analysis/signals.py
@@ -4,7 +4,7 @@
 import xml.etree.ElementTree as ET
 from pathlib import Path
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/analysis/temporal.py b/src/knowcode/analysis/temporal.py
@@ -6,7 +6,7 @@
 
 from git import Repo
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/cli/cli.py b/src/knowcode/cli/cli.py
@@ -8,7 +8,7 @@
 import click
 
 from knowcode import __version__
-from knowcode.models import EntityKind, RelationshipKind
+from knowcode.data_models import EntityKind, RelationshipKind
 from knowcode.service import KnowCodeService
 from knowcode.storage.knowledge_store import KnowledgeStore
 
diff --git a/src/knowcode/data_models.py b/src/knowcode/data_models.py
diff --git a/src/knowcode/indexing/chunker.py b/src/knowcode/indexing/chunker.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 from typing import Optional
 
-from knowcode.models import ChunkingConfig, CodeChunk, ParseResult, Entity, EntityKind
+from knowcode.data_models import ChunkingConfig, CodeChunk, ParseResult, Entity, EntityKind
 from knowcode.utils.tokenizer import tokenize_code
 from knowcode.utils.logger import get_logger
 
diff --git a/src/knowcode/indexing/graph_builder.py b/src/knowcode/indexing/graph_builder.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 from typing import Optional
 
-from knowcode.models import Entity, ParseResult, Relationship
+from knowcode.data_models import Entity, ParseResult, Relationship
 from knowcode.parsers import MarkdownParser, PythonParser, YamlParser
 from knowcode.parsers.javascript_parser import JavaScriptParser
 from knowcode.parsers.java_parser import JavaParser
diff --git a/src/knowcode/llm/embedding.py b/src/knowcode/llm/embedding.py
@@ -5,7 +5,7 @@
 from typing import Optional
 
 from openai import OpenAI
-from knowcode.models import EmbeddingConfig
+from knowcode.data_models import EmbeddingConfig
 
 
 class EmbeddingProvider(ABC):
diff --git a/src/knowcode/parsers/base.py b/src/knowcode/parsers/base.py
@@ -7,7 +7,7 @@
 from tree_sitter import Parser
 import tree_sitter_languages
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/parsers/java_parser.py b/src/knowcode/parsers/java_parser.py
@@ -3,7 +3,7 @@
 from pathlib import Path
 from typing import Any
 
-from knowcode.models import Entity, EntityKind, Relationship, RelationshipKind
+from knowcode.data_models import Entity, EntityKind, Relationship, RelationshipKind
 from knowcode.parsers.base import TreeSitterParser
 
 
diff --git a/src/knowcode/parsers/javascript_parser.py b/src/knowcode/parsers/javascript_parser.py
@@ -3,7 +3,7 @@
 from pathlib import Path
 from typing import Any
 
-from knowcode.models import Entity, EntityKind, Relationship, RelationshipKind
+from knowcode.data_models import Entity, EntityKind, Relationship, RelationshipKind
 from knowcode.parsers.base import TreeSitterParser
 
 
diff --git a/src/knowcode/parsers/markdown_parser.py b/src/knowcode/parsers/markdown_parser.py
@@ -5,7 +5,7 @@
 import re
 from pathlib import Path
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/parsers/python_parser.py b/src/knowcode/parsers/python_parser.py
@@ -7,7 +7,7 @@
 from pathlib import Path
 from typing import Optional
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/parsers/yaml_parser.py b/src/knowcode/parsers/yaml_parser.py
@@ -6,7 +6,7 @@
 
 import yaml
 
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/src/knowcode/retrieval/hybrid_index.py b/src/knowcode/retrieval/hybrid_index.py
@@ -3,7 +3,7 @@
 from typing import Optional
 
 from knowcode.storage.chunk_repository import ChunkRepository
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 from knowcode.utils.tokenizer import tokenize_code
 from knowcode.storage.vector_store import VectorStore
 
diff --git a/src/knowcode/retrieval/reranker.py b/src/knowcode/retrieval/reranker.py
@@ -2,7 +2,7 @@
 
 from typing import List, Tuple
 
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 
 
 class Reranker:
diff --git a/src/knowcode/retrieval/search_engine.py b/src/knowcode/retrieval/search_engine.py
@@ -7,7 +7,7 @@
 from knowcode.llm.embedding import EmbeddingProvider
 from knowcode.retrieval.hybrid_index import HybridIndex
 from knowcode.storage.knowledge_store import KnowledgeStore
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 from knowcode.retrieval.reranker import Reranker
 
 
diff --git a/src/knowcode/service.py b/src/knowcode/service.py
@@ -8,7 +8,7 @@
 from knowcode.analysis.context_synthesizer import ContextSynthesizer
 from knowcode.indexing.graph_builder import GraphBuilder
 from knowcode.storage.knowledge_store import KnowledgeStore
-from knowcode.models import EmbeddingConfig
+from knowcode.data_models import EmbeddingConfig
 
 
 class KnowCodeService:
diff --git a/src/knowcode/storage/chunk_repository.py b/src/knowcode/storage/chunk_repository.py
@@ -3,7 +3,7 @@
 from abc import ABC, abstractmethod
 from typing import Optional
 
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 
 
 class ChunkRepository(ABC):
diff --git a/src/knowcode/storage/knowledge_store.py b/src/knowcode/storage/knowledge_store.py
@@ -8,7 +8,7 @@
 from typing import Any, Optional
 
 from knowcode.indexing.graph_builder import GraphBuilder
-from knowcode.models import (
+from knowcode.data_models import (
     Entity,
     EntityKind,
     Location,
diff --git a/tests/integration/test_full_flow.py b/tests/integration/test_full_flow.py
@@ -6,7 +6,7 @@
 from knowcode.indexing.indexer import Indexer
 from knowcode.retrieval.search_engine import SearchEngine
 from knowcode.llm.embedding import OpenAIEmbeddingProvider
-from knowcode.models import EmbeddingConfig, CodeChunk
+from knowcode.data_models import EmbeddingConfig, CodeChunk
 from knowcode.retrieval.hybrid_index import HybridIndex
 from knowcode.storage.chunk_repository import InMemoryChunkRepository
 from knowcode.storage.vector_store import VectorStore
diff --git a/tests/integration/test_v2_1_integration.py b/tests/integration/test_v2_1_integration.py
@@ -3,7 +3,7 @@
 import pytest
 from unittest.mock import MagicMock
 from pathlib import Path
-from knowcode.models import EmbeddingConfig
+from knowcode.data_models import EmbeddingConfig
 from knowcode.llm.embedding import EmbeddingProvider
 from knowcode.indexing.indexer import Indexer
 from knowcode.storage.chunk_repository import InMemoryChunkRepository
diff --git a/tests/unit/analysis/test_signals.py b/tests/unit/analysis/test_signals.py
@@ -2,7 +2,7 @@
 
 import pytest
 from knowcode.analysis.signals import CoverageProcessor
-from knowcode.models import EntityKind, RelationshipKind
+from knowcode.data_models import EntityKind, RelationshipKind
 
 @pytest.fixture
 def coverage_xml(tmp_path):
diff --git a/tests/unit/analysis/test_temporal.py b/tests/unit/analysis/test_temporal.py
@@ -4,7 +4,7 @@
 from git import Repo, Actor
 
 from knowcode.analysis.temporal import TemporalAnalyzer
-from knowcode.models import EntityKind, RelationshipKind
+from knowcode.data_models import EntityKind, RelationshipKind
 
 @pytest.fixture
 def git_repo(tmp_path):
diff --git a/tests/unit/analysis/test_token_optimization.py b/tests/unit/analysis/test_token_optimization.py
@@ -3,7 +3,7 @@
 from unittest.mock import MagicMock
 from knowcode.utils.token_counter import TokenCounter
 from knowcode.analysis.context_synthesizer import ContextSynthesizer
-from knowcode.models import Entity, EntityKind, Location
+from knowcode.data_models import Entity, EntityKind, Location
 
 def test_token_counter():
     """Test functionality of TokenCounter."""
diff --git a/tests/unit/api/test_api_endpoints.py b/tests/unit/api/test_api_endpoints.py
@@ -4,7 +4,7 @@
 from fastapi.testclient import TestClient
 
 from knowcode.api import api
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 
 
 class DummySearchEngine:
diff --git a/tests/unit/indexing/test_chunker.py b/tests/unit/indexing/test_chunker.py
@@ -3,7 +3,7 @@
 from pathlib import Path
 
 from knowcode.indexing.chunker import Chunker
-from knowcode.models import ChunkingConfig, Entity, EntityKind, Location, ParseResult
+from knowcode.data_models import ChunkingConfig, Entity, EntityKind, Location, ParseResult
 
 
 def test_chunker_module_extraction() -> None:
diff --git a/tests/unit/indexing/test_graph_builder_references.py b/tests/unit/indexing/test_graph_builder_references.py
@@ -1,7 +1,7 @@
 """Unit tests for graph builder reference resolution."""
 
 from knowcode.indexing.graph_builder import GraphBuilder
-from knowcode.models import Entity, EntityKind, Location, Relationship, RelationshipKind
+from knowcode.data_models import Entity, EntityKind, Location, Relationship, RelationshipKind
 
 
 def test_reference_resolution() -> None:
diff --git a/tests/unit/llm/test_agent.py b/tests/unit/llm/test_agent.py
@@ -5,7 +5,7 @@
 
 import pytest
 from knowcode.llm.agent import Agent
-from knowcode.models import Entity, EntityKind, Location
+from knowcode.data_models import Entity, EntityKind, Location
 
 
 @pytest.fixture
diff --git a/tests/unit/llm/test_embedding_provider.py b/tests/unit/llm/test_embedding_provider.py
@@ -1,7 +1,7 @@
 """Unit tests for embedding providers."""
 
 from knowcode.llm.embedding import OpenAIEmbeddingProvider
-from knowcode.models import EmbeddingConfig
+from knowcode.data_models import EmbeddingConfig
 
 
 def test_embedding_provider_empty_batch() -> None:
diff --git a/tests/unit/models/test_models.py b/tests/unit/models/test_models.py
@@ -1,6 +1,6 @@
 """Unit tests for core models."""
 
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 
 
 def test_code_chunk_defaults() -> None:
diff --git a/tests/unit/parsers/test_java_parser.py b/tests/unit/parsers/test_java_parser.py
@@ -1,7 +1,7 @@
 """Tests for Java parser."""
 
 from pathlib import Path
-from knowcode.models import EntityKind, RelationshipKind
+from knowcode.data_models import EntityKind, RelationshipKind
 from knowcode.parsers.java_parser import JavaParser
 
 
diff --git a/tests/unit/parsers/test_javascript_parser.py b/tests/unit/parsers/test_javascript_parser.py
@@ -1,7 +1,7 @@
 """Tests for JavaScript parser."""
 
 from pathlib import Path
-from knowcode.models import EntityKind, RelationshipKind
+from knowcode.data_models import EntityKind, RelationshipKind
 from knowcode.parsers.javascript_parser import JavaScriptParser
 
 
diff --git a/tests/unit/retrieval/test_completeness.py b/tests/unit/retrieval/test_completeness.py
@@ -1,6 +1,6 @@
 """Unit tests for dependency expansion."""
 
-from knowcode.models import CodeChunk, Entity, EntityKind, Location, Relationship, RelationshipKind
+from knowcode.data_models import CodeChunk, Entity, EntityKind, Location, Relationship, RelationshipKind
 from knowcode.retrieval.completeness import expand_dependencies
 from knowcode.storage.chunk_repository import InMemoryChunkRepository
 from knowcode.storage.knowledge_store import KnowledgeStore
diff --git a/tests/unit/retrieval/test_hybrid_index.py b/tests/unit/retrieval/test_hybrid_index.py
@@ -1,6 +1,6 @@
 """Unit tests for hybrid retrieval scoring."""
 
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 from knowcode.retrieval.hybrid_index import HybridIndex
 
 
diff --git a/tests/unit/retrieval/test_search_engine.py b/tests/unit/retrieval/test_search_engine.py
@@ -1,6 +1,6 @@
 """Unit tests for search engine dependency expansion."""
 
-from knowcode.models import CodeChunk, Entity, EntityKind, Location, Relationship, RelationshipKind
+from knowcode.data_models import CodeChunk, Entity, EntityKind, Location, Relationship, RelationshipKind
 from knowcode.retrieval.search_engine import SearchEngine
 from knowcode.storage.chunk_repository import InMemoryChunkRepository
 from knowcode.storage.knowledge_store import KnowledgeStore
diff --git a/tests/unit/storage/test_chunk_repository.py b/tests/unit/storage/test_chunk_repository.py
@@ -1,6 +1,6 @@
 """Unit tests for chunk repositories."""
 
-from knowcode.models import CodeChunk
+from knowcode.data_models import CodeChunk
 from knowcode.storage.chunk_repository import InMemoryChunkRepository
 
 
diff --git a/tests/unit/storage/test_knowledge_store.py b/tests/unit/storage/test_knowledge_store.py
@@ -2,7 +2,7 @@
 
 import json
 
-from knowcode.models import Entity, EntityKind, Location, Relationship, RelationshipKind
+from knowcode.data_models import Entity, EntityKind, Location, Relationship, RelationshipKind
 from knowcode.storage.knowledge_store import KnowledgeStore
 
 
