chore: bump v0.8.16 (#2724)

Co-authored-by: Kian Jones <[email protected]>
Co-authored-by: Sarah Wooders <[email protected]>
Co-authored-by: Matthew Zhou <[email protected]>
Co-authored-by: Andy Li <[email protected]>
Co-authored-by: jnjpng <[email protected]>
Co-authored-by: Jin Peng <[email protected]>
Co-authored-by: cpacker <[email protected]>
Co-authored-by: Shubham Naik <[email protected]>
Co-authored-by: Shubham Naik <[email protected]>
Co-authored-by: Kevin Lin <[email protected]>
Co-authored-by: Eric Ly <[email protected]>
Co-authored-by: Eric Ly <[email protected]>

Author: 13 people

alembic/env.py

@@ -6,17 +6,17 @@
 from alembic import context
 from letta.config import LettaConfig
 from letta.orm import Base
-from letta.settings import settings
+from letta.settings import DatabaseChoice, settings
 
 letta_config = LettaConfig.load()
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.
 config = context.config
 
-if settings.letta_pg_uri_no_default:
+if settings.database_engine is DatabaseChoice.POSTGRES:
     config.set_main_option("sqlalchemy.url", settings.letta_pg_uri)
-    print(f"Using database: ", settings.letta_pg_uri)
+    print("Using database: ", settings.letta_pg_uri)
 else:
     config.set_main_option("sqlalchemy.url", "sqlite:///" + os.path.join(letta_config.recall_storage_path, "sqlite.db"))
 

alembic/versions/0335b1eb9c40_add_batch_item_id_to_messages.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "0335b1eb9c40"
@@ -20,12 +21,20 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.add_column("messages", sa.Column("batch_item_id", sa.String(), nullable=True))
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_column("messages", "batch_item_id")
     # ### end Alembic commands ###

alembic/versions/08b2f8225812_adding_toolsagents_orm.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "08b2f8225812"
@@ -20,6 +21,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_table(
         "tools_agents",
@@ -44,6 +49,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_table("tools_agents")
     # ### end Alembic commands ###

alembic/versions/0b496eae90de_add_file_agent_table.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "0b496eae90de"
@@ -20,6 +21,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_table(
         "files_agents",
@@ -48,6 +53,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_index("ix_files_agents_file_id_agent_id", table_name="files_agents")
     op.drop_table("files_agents")

alembic/versions/0ceb975e0063_add_llm_batch_jobs_tables.py

@@ -12,6 +12,7 @@
 
 import letta
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "0ceb975e0063"
@@ -21,6 +22,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_table(
         "llm_batch_job",
@@ -75,6 +80,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_index("ix_llm_batch_items_status", table_name="llm_batch_items")
     op.drop_index("ix_llm_batch_items_batch_id", table_name="llm_batch_items")

alembic/versions/167491cfb7a8_add_identities_for_blocks.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "167491cfb7a8"
@@ -20,6 +21,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_table(
         "identities_blocks",
@@ -33,6 +38,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_table("identities_blocks")
     # ### end Alembic commands ###

alembic/versions/18e300709530_add_instructions_field_to_sources.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "18e300709530"
@@ -20,12 +21,20 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.add_column("sources", sa.Column("instructions", sa.String(), nullable=True))
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_column("sources", "instructions")
     # ### end Alembic commands ###

alembic/versions/1af251a42c06_fix_files_agents_constraints.py

@@ -9,6 +9,7 @@
 from typing import Sequence, Union
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "1af251a42c06"
@@ -18,6 +19,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_index("ix_files_agents_agent_file_name", table_name="files_agents")
     op.drop_index("ix_files_agents_file_id_agent_id", table_name="files_agents")
@@ -31,6 +36,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_constraint("uq_file_agent", "files_agents", type_="unique")
     op.drop_constraint("uq_agent_filename", "files_agents", type_="unique")

alembic/versions/1c6b6a38b713_add_pip_requirements_to_tools.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "1c6b6a38b713"
@@ -20,12 +21,20 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.add_column("tools", sa.Column("pip_requirements", sa.JSON(), nullable=True))
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_column("tools", "pip_requirements")
     # ### end Alembic commands ###

alembic/versions/1c8880d671ee_make_an_blocks_agents_mapping_table.py

@@ -11,6 +11,7 @@
 import sqlalchemy as sa
 
 from alembic import op
+from letta.settings import settings
 
 # revision identifiers, used by Alembic.
 revision: str = "1c8880d671ee"
@@ -20,6 +21,10 @@
 
 
 def upgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.create_unique_constraint("unique_block_id_label", "block", ["id", "label"])
 
@@ -46,6 +51,10 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
+    # Skip this migration for SQLite
+    if not settings.letta_pg_uri_no_default:
+        return
+
     # ### commands auto generated by Alembic - please adjust! ###
     op.drop_constraint("unique_block_id_label", "block", type_="unique")
     op.drop_table("blocks_agents")