Add httpx

langchain-ai · Dec 13, 2023 · 0c2005e · 0c2005e
1 parent 646ce68
commit 0c2005e
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 6 deletions.
diff --git a/libs/community/langchain_community/llms/chatglm3.py b/libs/community/langchain_community/llms/chatglm3.py
@@ -3,10 +3,9 @@
 from typing import Any, List, Optional, Union
 
 import httpx
-from langchain.callbacks.manager import CallbackManagerForLLMRun
-from langchain.llms.base import LLM
-from langchain.llms.utils import enforce_stop_tokens
-from langchain.schema.messages import (
+from langchain_core.callbacks import CallbackManagerForLLMRun
+from langchain_core.language_models.llms import LLM
+from langchain_core.messages import (
     AIMessage,
     BaseMessage,
     FunctionMessage,
@@ -15,6 +14,8 @@
 )
 from langchain_core.pydantic_v1 import Field
 
+from langchain_community.llms.utils import enforce_stop_tokens
+
 logger = logging.getLogger(__name__)
 HEADERS = {"Content-Type": "application/json"}
 DEFAULT_TIMEOUT = 30

diff --git a/libs/community/poetry.lock b/libs/community/poetry.lock
diff --git a/libs/community/pyproject.toml b/libs/community/pyproject.toml
@@ -83,6 +83,7 @@ msal = {version = "^1.25.0", optional = true}
 databricks-vectorsearch = {version = "^0.21", optional = true}
 dgml-utils = {version = "^0.3.0", optional = true}
 datasets = {version = "^2.15.0", optional = true}
+httpx = "^0.24.1"
 
 [tool.poetry.group.test]
 optional = true

diff --git a/libs/community/tests/unit_tests/test_dependencies.py b/libs/community/tests/unit_tests/test_dependencies.py
@@ -41,6 +41,7 @@ def test_required_dependencies(poetry_conf: Mapping[str, Any]) -> None:
             "SQLAlchemy",
             "aiohttp",
             "dataclasses-json",
+            "httpx",
             "langchain-core",
             "langsmith",
             "numpy",

diff --git a/libs/langchain/poetry.lock b/libs/langchain/poetry.lock
diff --git a/libs/langchain/pyproject.toml b/libs/langchain/pyproject.toml
@@ -22,6 +22,7 @@ numpy = "^1"
 aiohttp = "^3.8.3"
 tenacity = "^8.1.0"
 jsonpatch = "^1.33"
+httpx = "^0.24.1"
 azure-core = {version = "^1.26.4", optional=true}
 tqdm = {version = ">=4.48.0", optional = true}
 openapi-pydantic = {version = "^0.3.2", optional = true}

diff --git a/libs/langchain/tests/unit_tests/test_dependencies.py b/libs/langchain/tests/unit_tests/test_dependencies.py
@@ -42,6 +42,7 @@ def test_required_dependencies(poetry_conf: Mapping[str, Any]) -> None:
             "aiohttp",
             "async-timeout",
             "dataclasses-json",
+            "httpx",
             "jsonpatch",
             "langchain-core",
             "langsmith",