Commit 7ee219bd authored by Jan Reimes's avatar Jan Reimes
Browse files

refactor(ai): remove unused workspace status and migration references 

* Removed WorkspaceStatus class from models.py as it is unnecessary.
* Eliminated references to consolidate_workspaces and migrate_to_shared_storage in __init__.py.
* Cleaned up specs_dir assignment in workspaces.py for clarity.
parent a43e7aa2

packages/3gpp-ai/threegpp_ai/lightrag/__init__.py

+0 −3
Original line number Diff line number Diff line
@@ -29,7 +29,6 @@ from .config import ( 
    StorageBackend,

)

from .metadata import RAGMetadata, create_metadata_from_dict, enrich_text

from .migration import consolidate_workspaces, migrate_to_shared_storage

from .pg0_manager import Pg0Error, Pg0Manager

from .processor import DocumentProcessor, ProcessingResult, ProcessingResultStatus, TDocProcessor

from .rag import TDocRAG
@@ -56,9 +55,7 @@ __all__ = [ 
    "TDocProcessor",

    "TDocRAG",

    "WorkspaceIndex",

    "consolidate_workspaces",

    "create_metadata_from_dict",

    "enrich_text",

    "initialize_shared_storage",

    "migrate_to_shared_storage",

]

packages/3gpp-ai/threegpp_ai/models.py

+0 −9
Original line number Diff line number Diff line
@@ -39,13 +39,6 @@ class GraphEdgeType(StrEnum): 
    REVISION_OF = "revision_of"  # is_revision_of metadata relationship





# TODO: Unnecessary? We can infer from presence of WorkspaceMember records and their is_active flag - or simply use a bool variable?

class WorkspaceStatus(StrEnum):

    """Lifecycle state of a workspace."""



    ACTIVE = "active"





class SourceKind(StrEnum):

    """Kinds of source items that can be part of a workspace corpus."""
@@ -90,7 +83,6 @@ class Workspace(BaseModel): 
    updated_at: datetime = Field(default_factory=utc_now, description="Last update timestamp")

    is_default: bool = Field(False, description="Whether this workspace is the default workspace")

    auto_build: bool | None = Field(default=False, description="Whether to auto-build embeddings on TDoc add")

    status: WorkspaceStatus = Field(WorkspaceStatus.ACTIVE, description="Workspace lifecycle status")



    @field_validator("workspace_name")

    @classmethod
@@ -401,5 +393,4 @@ __all__ = [ 
    "TDocNotFoundError",

    "Workspace",

    "WorkspaceMember",

    "WorkspaceStatus",

]

packages/3gpp-ai/threegpp_ai/operations/workspaces.py

+1 −1
Original line number Diff line number Diff line
@@ -539,7 +539,7 @@ async def checkout_spec_to_workspace( 
    requested_release = release.strip()



    # First check if already checked out with the SAME release version

    specs_dir = checkout_base / "Specs"  # TODO: resolve specs_dir from cache_manager!

    specs_dir = checkout_base / "Specs"

    if specs_dir.exists():

        # Resolve release to actual version(s) to match against 3GPP version codes (e.g., "h00", "i10")

        version_codes: list[str] = []