fix: patch tests (#1304)

.github/workflows/isort_format.yml

@@ -30,5 +30,5 @@ jobs:
         run: |
           poetry run isort --version
           poetry run python --version
-          echo "isort check failed. To fix, please run 'poetry run isort .'"
+          echo "isort check failed. To fix, please run 'poetry run isort --profile black .'"
           exit 1

memgpt/metadata.py

@@ -3,9 +3,9 @@
 import inspect as python_inspect
 import os
 import secrets
+import traceback
 import uuid
 from typing import List, Optional
-import traceback
 
 from sqlalchemy import (
     BIGINT,
@@ -20,9 +20,9 @@ from sqlalchemy import (
     func,
 )
 from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.exc import InterfaceError
 from sqlalchemy.orm import declarative_base, sessionmaker
 from sqlalchemy.sql import func
-from sqlalchemy.exc import InterfaceError
 
 from memgpt.config import MemGPTConfig
 from memgpt.data_types import (

memgpt/server/rest_api/tools/index.py

@@ -1,11 +1,9 @@
-from functools import partial
 from typing import List, Literal, Optional
 
 from fastapi import APIRouter, Body
 from pydantic import BaseModel, Field
 
 from memgpt.models.pydantic_models import ToolModel
-from memgpt.server.rest_api.auth_token import get_current_user
 from memgpt.server.rest_api.interface import QueuingInterface
 from memgpt.server.server import SyncServer
 
@@ -28,7 +26,7 @@ class CreateToolResponse(BaseModel):
 
 
 def setup_tools_index_router(server: SyncServer, interface: QueuingInterface, password: str):
-    partial(partial(get_current_user, server), password)
+    # get_current_user_with_server = partial(partial(get_current_user, server), password)
 
     @router.get("/tools", tags=["tools"], response_model=ListToolsResponse)
     async def list_all_tools(

tests/data/memgpt-0.2.11/agents/agent_test/config.json

@@ -10,7 +10,7 @@
     "model_wrapper": null,
     "embedding_endpoint_type": "openai",
     "embedding_endpoint": "https://api.openai.com/v1",
-    "embedding_model": null,
+    "embedding_model": "text-embedding-ada-002",
     "embedding_dim": 1536,
     "embedding_chunk_size": 300,
     "data_sources": [],

tests/data/memgpt-0.2.11/agents/agent_test_attach/config.json

@@ -10,7 +10,7 @@
     "model_wrapper": null,
     "embedding_endpoint_type": "openai",
     "embedding_endpoint": "https://api.openai.com/v1",
-    "embedding_model": null,
+    "embedding_model": "text-embedding-ada-002",
     "embedding_dim": 1536,
     "embedding_chunk_size": 300,
     "data_sources": [

tests/data/memgpt-0.2.11/agents/agent_test_empty_archival/config.json

@@ -10,7 +10,7 @@
     "model_wrapper": null,
     "embedding_endpoint_type": "openai",
     "embedding_endpoint": "https://api.openai.com/v1",
-    "embedding_model": null,
+    "embedding_model": "text-embedding-ada-002",
     "embedding_dim": 1536,
     "embedding_chunk_size": 300,
     "data_sources": [],