[feat]: weaviate db models alignment to support user indexing and search

Author: smokeyScraper

backend/app/model/weaviate/models.py

@@ -1,35 +1,88 @@
 from pydantic import BaseModel, Field
-from typing import List
+from typing import List, Optional
+from datetime import datetime
 
-
-class WeaviateUserProfile(BaseModel):
+class WeaviateRepository(BaseModel):
     """
-    Represents a vectorized user profile for semantic search in Weaviate.
+    Represents a single repostiory within WeaviateUserProfile.
+    Helps in structuring the repository-specific data that contributes to the user's overall profile
     """
-    supabase_user_id: str = Field(..., alias="supabaseUserId")
-    profile_summary: str = Field(..., alias="profileSummary")
-    primary_languages: List[str] = Field(..., alias="primaryLanguages")
-    expertise_areas: List[str] = Field(..., alias="expertiseAreas")
-    embedding: List[float] = Field(..., description="384-dimensional vector")
+    name: str = Field(..., description="The name of the repository.")
+    description: Optional[str] = Field(None, description="The repository's description.")
+    url: str = Field(..., description="The URL of the repository.")
+    languages: List[str] = Field(..., description="The languages used in the repository.")
+    stars: int = Field(0, description="The number of stars the repository has.")
+    forks: int = Field(0, description="The number of forks the repository has.")
 
 
-class WeaviateCodeChunk(BaseModel):
+class WeaviateUserProfile(BaseModel):
     """
-    Vectorized representation of code chunks stored in Weaviate.
+    Represents a user's profile data to be stored and indexed in Weaviate.
+    Enables semantic search capabilities to find users based on their profile data.
     """
-    supabase_chunk_id: str = Field(..., alias="supabaseChunkId")
-    code_content: str = Field(..., alias="codeContent")
-    language: str
-    function_names: List[str] = Field(..., alias="functionNames")
-    embedding: List[float] = Field(..., description="384-dimensional vector")
+    user_id: str = Field(..., description="The unique identifier for the user, linking back to the Supabase 'users' table.")
+    github_username: str = Field(..., description="The user's unique GitHub username.")
+    display_name: Optional[str] = Field(None, description="User's display name.")
+    bio: Optional[str] = Field(None, description="User's biography from their GitHub profile.")
+    location: Optional[str] = Field(None, description="User's location.")
 
+    repositories: List[WeaviateRepository] = Field(
+        default_factory=list, description="List of repositories the user's repositories.")
 
-class WeaviateInteraction(BaseModel):
-    """
-    Vectorized interaction representation stored in Weaviate.
-    """
-    supabase_interaction_id: str = Field(..., alias="supabaseInteractionId")
-    conversation_summary: str = Field(..., alias="conversationSummary")
-    platform: str
-    topics: List[str]
-    embedding: List[float] = Field(..., description="384-dimensional vector")
+    languages: List[str] = Field(default_factory=list,
+                                 description="A unique, aggregated list of all programming languages from the user's repositories.")
+    topics: List[str] = Field(default_factory=list,
+                              description="A unique, aggregated list of all topics from the user's repositories.")
+
+    followers_count: int = Field(0, description="Number of followers the user has on GitHub.")
+    following_count: int = Field(0, description="Number of other users this user is following on GitHub.")
+    total_stars_received: int = Field(
+        0, description="Total number of stars received across all of the user's owned repositories.")
+    total_forks: int = Field(0, description="Total number of times the user's repositories have been forked.")
+
+    profile_text_for_embedding: str = Field(
+        ..., description="A synthesized text field combining bio, repository names, descriptions, languages, and topics for vectorization.")
+
+    last_updated: datetime = Field(default_factory=datetime.now,
+                                   description="The date and time the profile was last updated.")
+
+    class Config:
+        """
+        Pydantic model configuration.
+        """
+        orm_mode = True
+        schema_extra = {
+            "example": {
+                "user_id": "a1b2c3d4-e5f6-7890-1234-567890abcdef",
+                "github_username": "jane-dev",
+                "display_name": "Jane Developer",
+                "bio": "Creator of innovative open-source tools. Full-stack developer with a passion for Rust and WebAssembly.",
+                "location": "Berlin, Germany",
+                "repositories": [
+                    {
+                        "name": "rust-web-framework",
+                        "description": "A high-performance web framework for Rust.",
+                        "languages": ["Rust", "TOML"],
+                        "topics": ["rust", "webdev", "performance", "framework"],
+                        "stars": 2500,
+                        "forks": 400
+                    },
+                    {
+                        "name": "data-viz-lib",
+                        "description": "A declarative data visualization library for JavaScript.",
+                        "languages": ["JavaScript", "TypeScript"],
+                        "topics": ["data-visualization", "d3", "charts"],
+                        "stars": 1200,
+                        "forks": 150
+                    }
+                ],
+                "languages": ["Rust", "JavaScript", "TypeScript", "TOML"],
+                "topics": ["rust", "webdev", "performance", "framework", "data-visualization", "d3", "charts"],
+                "followers_count": 1800,
+                "following_count": 250,
+                "total_stars_received": 3700,
+                "total_forks": 550,
+                "profile_text_for_embedding": "Jane Developer, Creator of innovative open-source tools. Full-stack developer with a passion for Rust and WebAssembly. Repositories: rust-web-framework, A high-performance web framework for Rust. data-viz-lib, A declarative data visualization library for JavaScript. Languages: Rust, JavaScript, TypeScript. Topics: rust, webdev, performance, data-visualization.",
+                "last_updated": "2025-06-23T12:21:00Z"
+            }
+        }