Add OpenAI proxy (#1503)

Co-authored-by: Deshraj Yadav <deshrajdry@gmail.com>

mem0/memory/main.py

@@ -15,51 +15,17 @@ from mem0.llms.utils.tools import (
 )
 from mem0.configs.prompts import MEMORY_DEDUCTION_PROMPT
 from mem0.memory.base import MemoryBase
-from mem0.memory.setup import mem0_dir, setup_config
+from mem0.memory.setup import setup_config
 from mem0.memory.storage import SQLiteManager
 from mem0.memory.telemetry import capture_event
 from mem0.memory.utils import get_update_memory_messages
-from mem0.vector_stores.configs import VectorStoreConfig
-from mem0.llms.configs import LlmConfig
-from mem0.embeddings.configs import EmbedderConfig
 from mem0.utils.factory import LlmFactory, EmbedderFactory, VectorStoreFactory
+from mem0.configs.base import MemoryItem, MemoryConfig
 
 # Setup user config
 setup_config()
 
 
-class MemoryItem(BaseModel):
-    id: str = Field(..., description="The unique identifier for the text data")
-    memory: str = Field(..., description="The memory deduced from the text data") # TODO After prompt changes from platform, update this
-    hash: Optional[str] = Field(None, description="The hash of the memory")
-    # The metadata value can be anything and not just string. Fix it
-    metadata: Optional[Dict[str, Any]] = Field(None, description="Additional metadata for the text data")
-    score: Optional[float] = Field(
-        None, description="The score associated with the text data"
-    )
-    created_at: Optional[str] = Field(None, description="The timestamp when the memory was created")
-    updated_at: Optional[str] = Field(None, description="The timestamp when the memory was updated")
-
-
-class MemoryConfig(BaseModel):
-    vector_store: VectorStoreConfig = Field(
-        description="Configuration for the vector store",
-        default_factory=VectorStoreConfig,
-    )
-    llm: LlmConfig = Field(
-        description="Configuration for the language model",
-        default_factory=LlmConfig,
-    )
-    embedder: EmbedderConfig = Field(
-        description="Configuration for the embedding model",
-        default_factory=EmbedderConfig,
-    )
-    history_db_path: str = Field(
-        description="Path to the history database",
-        default=os.path.join(mem0_dir, "history.db"),
-    )
-
-
 class Memory(MemoryBase):
     def __init__(self, config: MemoryConfig = MemoryConfig()):
         self.config = config

mem0/memory/storage.py

@@ -17,33 +17,52 @@ class SQLiteManager:
             table_exists = cursor.fetchone() is not None
 
             if table_exists:
-                # Rename the old table
-                cursor.execute("ALTER TABLE history RENAME TO old_history")
+                # Get the current schema of the history table
+                cursor.execute("PRAGMA table_info(history)")
+                current_schema = {row[1]: row[2] for row in cursor.fetchall()}
 
-                cursor.execute("""
-                    CREATE TABLE IF NOT EXISTS history (
-                        id TEXT PRIMARY KEY,
-                        memory_id TEXT,
-                        old_memory TEXT,
-                        new_memory TEXT,
-                        new_value TEXT,
-                        event TEXT,
-                        created_at DATETIME,
-                        updated_at DATETIME,
-                        is_deleted INTEGER
-                    )
-                """)
+                # Define the expected schema
+                expected_schema = {
+                    'id': 'TEXT',
+                    'memory_id': 'TEXT',
+                    'old_memory': 'TEXT',
+                    'new_memory': 'TEXT',
+                    'new_value': 'TEXT',
+                    'event': 'TEXT',
+                    'created_at': 'DATETIME',
+                    'updated_at': 'DATETIME',
+                    'is_deleted': 'INTEGER'
+                }
 
-                # Copy data from the old table to the new table
-                cursor.execute("""
-                    INSERT INTO history (id, memory_id, old_memory, new_memory, new_value, event, created_at, updated_at, is_deleted)
-                    SELECT id, memory_id, prev_value, new_value, new_value, event, timestamp, timestamp, is_deleted
-                    FROM old_history
-                """)
+                # Check if the schemas are the same
+                if current_schema != expected_schema:
+                    # Rename the old table
+                    cursor.execute("ALTER TABLE history RENAME TO old_history")
 
-                cursor.execute("DROP TABLE old_history")
+                    cursor.execute("""
+                        CREATE TABLE IF NOT EXISTS history (
+                            id TEXT PRIMARY KEY,
+                            memory_id TEXT,
+                            old_memory TEXT,
+                            new_memory TEXT,
+                            new_value TEXT,
+                            event TEXT,
+                            created_at DATETIME,
+                            updated_at DATETIME,
+                            is_deleted INTEGER
+                        )
+                    """)
 
-                self.connection.commit()
+                    # Copy data from the old table to the new table
+                    cursor.execute("""
+                        INSERT INTO history (id, memory_id, old_memory, new_memory, new_value, event, created_at, updated_at, is_deleted)
+                        SELECT id, memory_id, prev_value, new_value, new_value, event, timestamp, timestamp, is_deleted
+                        FROM old_history
+                    """)
+
+                    cursor.execute("DROP TABLE old_history")
+
+                    self.connection.commit()
 
 
     def _create_history_table(self):