diff --git a/libs/cli/langchain_cli/namespaces/migrate/codemods/migrations/astradb.json b/libs/cli/langchain_cli/namespaces/migrate/codemods/migrations/astradb.json new file mode 100644 index 00000000000..5a3f9c9849a --- /dev/null +++ b/libs/cli/langchain_cli/namespaces/migrate/codemods/migrations/astradb.json @@ -0,0 +1,30 @@ +[ + [ + "langchain_community.vectorstores.astradb.AstraDB", + "langchain_astradb.AstraDBVectorStore" + ], + [ + "langchain_community.storage.astradb.AstraDBByteStore", + "langchain_astradb.AstraDBByteStore" + ], + [ + "langchain_community.storage.astradb.AstraDBStore", + "langchain_astradb.AstraDBStore" + ], + [ + "langchain_community.cache.astradb.AstraDBCache", + "langchain_astradb.AstraDBCache" + ], + [ + "langchain_community.cache.astradb.AstraDBSemanticCache", + "langchain_astradb.AstraDBSemanticCache" + ], + [ + "langchain_community.chat_message_histories.astradb.AstraDBChatMessageHistory", + "langchain_astradb.AstraDBChatMessageHistory" + ], + [ + "langchain_community.document_loaders.astradb.AstraDBLoader", + "langchain_astradb.AstraDBLoader" + ] +] \ No newline at end of file diff --git a/libs/cli/langchain_cli/namespaces/migrate/codemods/replace_imports.py b/libs/cli/langchain_cli/namespaces/migrate/codemods/replace_imports.py index 11bf2025f0f..b6ba8913578 100644 --- a/libs/cli/langchain_cli/namespaces/migrate/codemods/replace_imports.py +++ b/libs/cli/langchain_cli/namespaces/migrate/codemods/replace_imports.py @@ -147,6 +147,7 @@ RULE_TO_PATHS = { "ibm.json", "openai.json", "pinecone.json", + "astradb.json", ], }