letta-nightly 0.5.2.dev20241107104040__tar.gz → 0.5.2.dev20241108104049__tar.gz

{letta_nightly-0.5.2.dev20241107104040 → letta_nightly-0.5.2.dev20241108104049}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-nightly
-Version: 0.5.2.dev20241107104040
+Version: 0.5.2.dev20241108104049
 Summary: Create LLM agents with long-term memory and custom tools
 License: Apache License
 Author: Letta Team

letta_nightly-0.5.2.dev20241108104049/letta/server/generate_openapi_schema.sh

@@ -0,0 +1,12 @@
+#!/bin/sh
+echo "Generating OpenAPI schema..."
+
+# check if poetry is installed
+if ! command -v poetry &> /dev/null
+then
+    echo "Poetry could not be found. Please install poetry to generate the OpenAPI schema."
+    exit
+fi
+
+# generate OpenAPI schema
+poetry run python -c 'from letta.server.rest_api.app import app, generate_openapi_schema; generate_openapi_schema(app);'

{letta_nightly-0.5.2.dev20241107104040 → letta_nightly-0.5.2.dev20241108104049}/letta/server/rest_api/app.py

@@ -63,6 +63,37 @@ from fastapi import FastAPI
 log = logging.getLogger("uvicorn")
 
 
+def generate_openapi_schema(app: FastAPI):
+    # Update the OpenAPI schema
+    if not app.openapi_schema:
+        app.openapi_schema = app.openapi()
+
+    openai_docs, letta_docs = [app.openapi_schema.copy() for _ in range(2)]
+
+    openai_docs["paths"] = {k: v for k, v in openai_docs["paths"].items() if k.startswith("/openai")}
+    openai_docs["info"]["title"] = "OpenAI Assistants API"
+    letta_docs["paths"] = {k: v for k, v in letta_docs["paths"].items() if not k.startswith("/openai")}
+    letta_docs["info"]["title"] = "Letta API"
+    letta_docs["components"]["schemas"]["LettaResponse"] = {
+        "properties": LettaResponse.model_json_schema(ref_template="#/components/schemas/LettaResponse/properties/{model}")["$defs"]
+    }
+
+    # Split the API docs into Letta API, and OpenAI Assistants compatible API
+    for name, docs in [
+        (
+            "openai",
+            openai_docs,
+        ),
+        (
+            "letta",
+            letta_docs,
+        ),
+    ]:
+        if settings.cors_origins:
+            docs["servers"] = [{"url": host} for host in settings.cors_origins]
+        Path(f"openapi_{name}.json").write_text(json.dumps(docs, indent=2))
+
+
 def create_application() -> "FastAPI":
     """the application start routine"""
     # global server
@@ -122,34 +153,7 @@ def create_application() -> "FastAPI":
 
         # Tool.load_default_tools(get_db_session())
 
-        # Update the OpenAPI schema
-        if not app.openapi_schema:
-            app.openapi_schema = app.openapi()
-
-        openai_docs, letta_docs = [app.openapi_schema.copy() for _ in range(2)]
-
-        openai_docs["paths"] = {k: v for k, v in openai_docs["paths"].items() if k.startswith("/openai")}
-        openai_docs["info"]["title"] = "OpenAI Assistants API"
-        letta_docs["paths"] = {k: v for k, v in letta_docs["paths"].items() if not k.startswith("/openai")}
-        letta_docs["info"]["title"] = "Letta API"
-        letta_docs["components"]["schemas"]["LettaResponse"] = {
-            "properties": LettaResponse.model_json_schema(ref_template="#/components/schemas/LettaResponse/properties/{model}")["$defs"]
-        }
-
-        # Split the API docs into Letta API, and OpenAI Assistants compatible API
-        for name, docs in [
-            (
-                "openai",
-                openai_docs,
-            ),
-            (
-                "letta",
-                letta_docs,
-            ),
-        ]:
-            if settings.cors_origins:
-                docs["servers"] = [{"url": host} for host in settings.cors_origins]
-            Path(f"openapi_{name}.json").write_text(json.dumps(docs, indent=2))
+        generate_openapi_schema(app)
 
     @app.on_event("shutdown")
     def on_shutdown():

letta_nightly-0.5.2.dev20241108104049/letta/server/startup.sh

@@ -0,0 +1,17 @@
+#!/bin/sh
+echo "Starting Letta server at http://localhost:8283"
+
+# Check if LETTA_PG_URI or LETTA_PG_DB is set and run alembic upgrade if either is
+if [ -n "$LETTA_PG_URI" ] || [ -n "$LETTA_PG_DB" ]; then
+    echo "LETTA_PG_URI or LETTA_PG_DB is set, running alembic upgrade head"
+    alembic upgrade head
+fi
+
+if [ "$MEMGPT_ENVIRONMENT" = "DEVELOPMENT" ]; then
+    echo "Starting in development mode!"
+    uvicorn letta.server.rest_api.app:app --reload --reload-dir /letta --host 0.0.0.0 --port 8283
+else
+    # Production start command here (replace with the actual production command)
+    echo "Starting in production mode!"
+    uvicorn letta.server.rest_api.app:app --host 0.0.0.0 --port 8283
+fi

{letta_nightly-0.5.2.dev20241107104040 → letta_nightly-0.5.2.dev20241108104049}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "letta-nightly"
-version = "0.5.2.dev20241107104040"
+version = "0.5.2.dev20241108104049"
 packages = [
     {include = "letta"}
 ]

letta_nightly-0.5.2.dev20241107104040/letta/server/startup.sh

@@ -1,11 +0,0 @@
-#!/bin/sh
-echo "Starting MEMGPT server..."
-
-alembic upgrade head
-
-if [ "$MEMGPT_ENVIRONMENT" = "DEVELOPMENT" ] ; then
-    echo "Starting in development mode!"
-    uvicorn letta.server.rest_api.app:app --reload --reload-dir /letta --host 0.0.0.0 --port 8283
-else
-    uvicorn letta.server.rest_api.app:app --host 0.0.0.0 --port 8283
-fi