From 2bfc657e90c3185a082b3c92d76f83c85849fbe4 Mon Sep 17 00:00:00 2001 From: vasilije Date: Sun, 5 Jan 2025 20:30:34 +0100 Subject: [PATCH] Implement PR review --- cognee/infrastructure/engine/models/DataPoint.py | 9 +++------ cognee/infrastructure/llm/openai/adapter.py | 3 --- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/cognee/infrastructure/engine/models/DataPoint.py b/cognee/infrastructure/engine/models/DataPoint.py index 9cc818228..0f2b0d34f 100644 --- a/cognee/infrastructure/engine/models/DataPoint.py +++ b/cognee/infrastructure/engine/models/DataPoint.py @@ -19,11 +19,9 @@ class DataPoint(BaseModel): id: UUID = Field(default_factory=uuid4) created_at: int = Field(default_factory=lambda: int(datetime.now(timezone.utc).timestamp() * 1000)) updated_at: int = Field(default_factory=lambda: int(datetime.now(timezone.utc).timestamp() * 1000)) - version: str = "0.1" # Default version - source: Optional[str] = None # Path to file, URL, etc. + version: str = "1" # Default version type: Optional[str] = "text" # "text", "file", "image", "video" topological_rank: Optional[int] = 0 - extra: Optional[str] = "extra" # For additional properties _metadata: Optional[MetaData] = { "index_fields": [], "type": "DataPoint" @@ -33,7 +31,6 @@ class DataPoint(BaseModel): class Config: underscore_attrs_are_private = True - @classmethod @classmethod def get_embeddable_data(self, data_point): if data_point._metadata and len(data_point._metadata["index_fields"]) > 0 \ @@ -56,9 +53,9 @@ class DataPoint(BaseModel): """Retrieve names of embeddable properties.""" return data_point._metadata["index_fields"] or [] - def update_version(self, new_version: str): + def update_version(self): """Update the version and updated_at timestamp.""" - self.version = new_version + self.version += 1 self.updated_at = int(datetime.now(timezone.utc).timestamp() * 1000) # JSON Serialization diff --git a/cognee/infrastructure/llm/openai/adapter.py b/cognee/infrastructure/llm/openai/adapter.py index a6bccdf7e..835096007 100644 --- a/cognee/infrastructure/llm/openai/adapter.py +++ b/cognee/infrastructure/llm/openai/adapter.py @@ -38,9 +38,6 @@ class OpenAIAdapter(LLMInterface): self.streaming = streaming base_config = get_base_config() if base_config.monitoring_tool == MonitoringTool.LANGFUSE: - # set callbacks - # litellm.success_callback = ["langfuse"] - # litellm.failure_callback = ["langfuse"] self.aclient.success_callback = ["langfuse"] self.aclient.failure_callback = ["langfuse"] self.client.success_callback = ["langfuse"]