minor fix: rename kgenrichmentresponse

py/core/base/api/models/__init__.py

@@ -34,8 +34,8 @@
     WrappedUserOverviewResponse,
 )
 from .restructure.responses import (
-    KGEnrichementResponse,
-    WrappedKGEnrichementResponse,
+    KGEnrichmentResponse,
+    WrappedKGEnrichmentResponse,
 )
 from .retrieval.responses import (
     RAGAgentResponse,
@@ -57,8 +57,8 @@
     "IngestionResponse",
     "WrappedIngestionResponse",
     # Restructure Responses
-    "KGEnrichementResponse",
-    "WrappedKGEnrichementResponse",
+    "KGEnrichmentResponse",
+    "WrappedKGEnrichmentResponse",
     # Management Responses
     "PromptResponse",
     "ServerStats",

py/core/base/api/models/restructure/responses.py

@@ -6,8 +6,8 @@
 from pydantic import BaseModel
 
 
-class KGEnrichementResponse(BaseModel):
+class KGEnrichmentResponse(BaseModel):
     enriched_content: Dict[str, Any]
 
 
-WrappedKGEnrichementResponse = ResultsWrapper[KGEnrichementResponse]
+WrappedKGEnrichmentResponse = ResultsWrapper[KGEnrichmentResponse]

py/sdk/models.py

@@ -88,17 +88,6 @@ class KGSearchSettings(BaseModel):
     }
 
 
-class KGEnrichmentSettings(BaseModel):
-    leiden_params: dict = Field(
-        default_factory=dict,
-        description="Parameters for the Leiden algorithm.",
-    )
-    generation_config: GenerationConfig = Field(
-        default_factory=GenerationConfig,
-        description="Configuration for text generation during graph enrichment.",
-    )
-
-
 class ProviderConfig(BaseModel, ABC):
     """A base provider configuration class"""
 
@@ -263,7 +252,7 @@ class KGEnrichmentSettings(BaseModel):
         description="The parameters for the Leiden algorithm.",
     )
 
-class KGEnrichementResponse(BaseModel):
+class KGEnrichmentResponse(BaseModel):
     enriched_content: Dict[str, Any]
 
 class UserResponse(BaseModel):

py/sdk/restructure.py

@@ -1,18 +1,18 @@
 from typing import Any, Dict, List, Optional, Union
 
 from core.base import Document
-from .models import KGEnrichmentSettings, KGEnrichementResponse
+from .models import KGEnrichmentSettings, KGEnrichmentResponse
 
 class RestructureMethods:
     @staticmethod
     async def enrich_graph(
         client, KGEnrichmentSettings: KGEnrichmentSettings = KGEnrichmentSettings()
-    ) -> KGEnrichementResponse:
+    ) -> KGEnrichmentResponse:
         """
         Perform graph enrichment over the entire graph.
 
         Returns:
-            KGEnrichementResponse: Results of the graph enrichment process.
+            KGEnrichmentResponse: Results of the graph enrichment process.
         """
         if not isinstance(KGEnrichmentSettings, dict):
             KGEnrichmentSettings = KGEnrichmentSettings.model_dump()