diff --git a/packages/gen/docs/gen_ai_hub/examples/ai-vs-ai.py b/packages/gen/docs/gen_ai_hub/examples/ai-vs-ai.py index 56eb3ab9..6d9cd0de 100644 --- a/packages/gen/docs/gen_ai_hub/examples/ai-vs-ai.py +++ b/packages/gen/docs/gen_ai_hub/examples/ai-vs-ai.py @@ -6,8 +6,7 @@ ChatPromptTemplate, MessagesPlaceholder, ) -from langchain_community.chat_message_histories.in_memory import ChatMessageHistory -from langchain_core.chat_history import BaseChatMessageHistory +from langchain_core.chat_history import BaseChatMessageHistory, InMemoryChatMessageHistory from langchain_core.messages import HumanMessage from langchain_core.runnables.history import RunnableWithMessageHistory @@ -37,13 +36,13 @@ def get_session_history(session_id: str) -> BaseChatMessageHistory: if session_id not in store: - store[session_id] = ChatMessageHistory() + store[session_id] = InMemoryChatMessageHistory() return store[session_id] def create_session_history() -> str: session_id = str(uuid.uuid4()) - store[session_id] = ChatMessageHistory() + store[session_id] = InMemoryChatMessageHistory() return session_id diff --git a/packages/gen/gen_ai_hub/proxy/langchain/amazon.py b/packages/gen/gen_ai_hub/proxy/langchain/amazon.py index 9f3d92cf..d67dbd7f 100644 --- a/packages/gen/gen_ai_hub/proxy/langchain/amazon.py +++ b/packages/gen/gen_ai_hub/proxy/langchain/amazon.py @@ -2,8 +2,7 @@ from typing import Dict, Optional, List, Any from botocore.config import Config -from langchain_aws import ChatBedrock as ChatBedrock_, ChatBedrockConverse as ChatBedrockConverse_ -from langchain_community.embeddings import BedrockEmbeddings as BedrockEmbeddings_ +from langchain_aws import ChatBedrock as ChatBedrock_, ChatBedrockConverse as ChatBedrockConverse_, BedrockEmbeddings as BedrockEmbeddings_ from pydantic import BaseModel, ConfigDict, model_validator from gen_ai_hub.proxy.core.base import BaseProxyClient diff --git a/packages/gen/integration_tests/langchain_/test_google_genai.py b/packages/gen/integration_tests/langchain_/test_google_genai.py index ab42cfca..c42fb60b 100644 --- a/packages/gen/integration_tests/langchain_/test_google_genai.py +++ b/packages/gen/integration_tests/langchain_/test_google_genai.py @@ -7,8 +7,8 @@ ChatPromptTemplate, MessagesPlaceholder, ) -from langchain_classic.schema import BaseChatMessageHistory, HumanMessage -from langchain_community.chat_message_histories.in_memory import ChatMessageHistory +from langchain_classic.schema import HumanMessage +from langchain_core.chat_history import BaseChatMessageHistory, InMemoryChatMessageHistory from langchain_core.messages import AIMessage from langchain_core.runnables.history import RunnableWithMessageHistory @@ -30,7 +30,7 @@ def _get_model_history(self, session_id: str) -> BaseChatMessageHistory: def _create_model_history(self) -> str: session_id = str(uuid.uuid4()) - self.model_histories[session_id] = ChatMessageHistory() + self.model_histories[session_id] = InMemoryChatMessageHistory() return session_id def test_genai_invoke(self, model=GEMINI_2_5_FLASH_LITE_TEST_MODEL): diff --git a/packages/gen/pyproject.toml b/packages/gen/pyproject.toml index dba3c984..b030f71b 100644 --- a/packages/gen/pyproject.toml +++ b/packages/gen/pyproject.toml @@ -45,7 +45,6 @@ dependencies = [ "pandas>=2.2.0", "langchain~=1.4.0", "langchain-classic~=1.0.0", - "langchain-community~=0.4.1", "langchain-openai~=1.6.0", ] diff --git a/packages/gen/tests/proxy/langchain_/test_amazon.py b/packages/gen/tests/proxy/langchain_/test_amazon.py index c3eac9d8..67477555 100644 --- a/packages/gen/tests/proxy/langchain_/test_amazon.py +++ b/packages/gen/tests/proxy/langchain_/test_amazon.py @@ -55,7 +55,7 @@ def test_chat_model(self, mock_chain_invoke): response = chain.invoke("I love programming") self.assertIsInstance(response["text"], str) - @patch("langchain_community.embeddings.bedrock.BedrockEmbeddings.embed_query") + @patch("langchain_aws.embeddings.bedrock.BedrockEmbeddings.embed_query") def test_embedding_model(self, mock_chain_invoke): mock_chain_invoke.return_value = AMAZON_TITAN_EMBED_QUERY_RESPONSE embedding_model = BedrockEmbeddings( diff --git a/uv.lock b/uv.lock index 558b55b7..86122e41 100644 --- a/uv.lock +++ b/uv.lock @@ -1269,15 +1269,6 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/2a/39/e50c7c3a983047577ee07d2a9e53faf5a69493943ec3f6a384bdc792deb2/httpx-0.28.1-py3-none-any.whl", hash = "sha256:d909fcccc110f8c7faf814ca82a9a4d816bc5a6dbfea25d6591d6985b8ba59ad", size = 73517, upload-time = "2024-12-06T15:37:21.509Z" }, ] -[[package]] -name = "httpx-sse" -version = "0.4.3" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/0f/4c/751061ffa58615a32c31b2d82e8482be8dd4a89154f003147acee90f2be9/httpx_sse-0.4.3.tar.gz", hash = "sha256:9b1ed0127459a66014aec3c56bebd93da3c1bc8bb6618c8082039a44889a755d", size = 15943, upload-time = "2025-10-10T21:48:22.271Z" } -wheels = [ - { url = "https://files.pythonhosted.org/packages/d2/fd/6668e5aec43ab844de6fc74927e155a3b37bf40d7c3790e49fc0406b6578/httpx_sse-0.4.3-py3-none-any.whl", hash = "sha256:0ac1c9fe3c0afad2e0ebb25a934a59f4c7823b60792691f779fad2c5568830fc", size = 8960, upload-time = "2025-10-10T21:48:21.158Z" }, -] - [[package]] name = "idna" version = "3.18" @@ -1705,30 +1696,6 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/99/9a/b8f5cb7490fdbf233088031fc69c9c747439d4097f67f196c1eb4869916d/langchain_classic-1.0.8-py3-none-any.whl", hash = "sha256:1a11ea7fbe630c4f2af2f3873d27718ceac9488cf32d0821030be7cf039a6213", size = 1041536, upload-time = "2026-06-10T21:27:52.767Z" }, ] -[[package]] -name = "langchain-community" -version = "0.4.2" -source = { registry = "https://pypi.org/simple" } -dependencies = [ - { name = "aiohttp" }, - { name = "httpx-sse" }, - { name = "langchain-classic" }, - { name = "langchain-core" }, - { name = "langsmith" }, - { name = "numpy", version = "2.2.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" }, - { name = "numpy", version = "2.4.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version == '3.11.*'" }, - { name = "numpy", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.12'" }, - { name = "pydantic-settings" }, - { name = "pyyaml" }, - { name = "requests" }, - { name = "sqlalchemy" }, - { name = "tenacity" }, -] -sdist = { url = "https://files.pythonhosted.org/packages/ea/0c/e3aca1f2b1c5b95f8b87cb2b6e81a6f20d538c07a128419dc01cef0617b6/langchain_community-0.4.2.tar.gz", hash = "sha256:a99308160d53d7e9b5965ee665e5173709914338210089fd5788ad724432c21e", size = 33268708, upload-time = "2026-05-22T19:42:59.374Z" } -wheels = [ - { url = "https://files.pythonhosted.org/packages/8f/39/5d97e42a3e95dc2a6d71b2f902a3fae71786131e11d01bddb604accb0ebe/langchain_community-0.4.2-py3-none-any.whl", hash = "sha256:84dd8c5122532394d5b6849a5fc9995ef28e4f77227daeb09f24b3d942e9e466", size = 2364406, upload-time = "2026-05-22T19:42:57.103Z" }, -] - [[package]] name = "langchain-core" version = "1.6.2" @@ -3350,20 +3317,6 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/4b/2d/69abac8f838090bbecd5df894befb2c2619e7996a98ddb949db9f3b93225/pydantic_core-2.46.4-pp311-pypy311_pp73-win_amd64.whl", hash = "sha256:d51026d73fcfd93610abc7b27789c26b313920fcfb20e27462d74a7f8b06e983", size = 2193071, upload-time = "2026-05-06T13:38:08.682Z" }, ] -[[package]] -name = "pydantic-settings" -version = "2.14.2" -source = { registry = "https://pypi.org/simple" } -dependencies = [ - { name = "pydantic" }, - { name = "python-dotenv" }, - { name = "typing-inspection" }, -] -sdist = { url = "https://files.pythonhosted.org/packages/5c/b5/8f48e906c3e0205276e8bd8cb7512217a87b2685304d64be27cad5b3019f/pydantic_settings-2.14.2.tar.gz", hash = "sha256:c19dd64b19097f1de80184f0cc7b0272a13ae6e170cbf240a3e27e381ed14a5f", size = 237700, upload-time = "2026-06-19T13:44:56.324Z" } -wheels = [ - { url = "https://files.pythonhosted.org/packages/77/c1/6e422f34e569cf8e18df68d1939c81c099d2b61e4f7d9621c8a77560799c/pydantic_settings-2.14.2-py3-none-any.whl", hash = "sha256:a20c97b37910b6550d5ea50fbcc2d4187defe58cd57070b73863d069419c9440", size = 61715, upload-time = "2026-06-19T13:44:55.02Z" }, -] - [[package]] name = "pygments" version = "2.20.0" @@ -4224,7 +4177,6 @@ dependencies = [ { name = "httpx" }, { name = "langchain" }, { name = "langchain-classic" }, - { name = "langchain-community" }, { name = "langchain-openai" }, { name = "langcodes" }, { name = "openai" }, @@ -4283,7 +4235,6 @@ requires-dist = [ { name = "langchain", specifier = "~=1.4.0" }, { name = "langchain-aws", marker = "extra == 'amazon'", specifier = "~=1.7.0" }, { name = "langchain-classic", specifier = "~=1.0.0" }, - { name = "langchain-community", specifier = "~=0.4.1" }, { name = "langchain-google-genai", marker = "extra == 'google'", specifier = "~=4.3.6,<4.3.7" }, { name = "langchain-openai", specifier = "~=1.6.0" }, { name = "langcodes", specifier = "~=3.5.1" },