github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-07 13:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Use Union instead of | to improve compatibility, fix #15244 (#15245)

Browse Source
This commit is contained in:
chyroc 2023-12-28 14:06:42 +08:00 committed by GitHub
parent 6a5a2fb9c8
commit 6fb3cc6f27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/chat_models/gpt_router.py
View File

@ -233,8 +233,8 @@ class GPTRouter(BaseChatModel):
self, self,
messages: List[BaseMessage], messages: List[BaseMessage],
stop: Optional[List[str]] = None, stop: Optional[List[str]] = None,
run_manager: CallbackManagerForLLMRun | None = None, run_manager: Optional[CallbackManagerForLLMRun] = None,
stream: bool | None = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> ChatResult:
should_stream = stream if stream is not None else self.streaming should_stream = stream if stream is not None else self.streaming
@ -259,8 +259,8 @@ class GPTRouter(BaseChatModel):
self, self,
messages: List[BaseMessage], messages: List[BaseMessage],
stop: Optional[List[str]] = None, stop: Optional[List[str]] = None,
run_manager: AsyncCallbackManagerForLLMRun | None = None, run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
stream: bool | None = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> ChatResult:
should_stream = stream if stream is not None else self.streaming should_stream = stream if stream is not None else self.streaming

6
libs/community/langchain_community/tools/e2b_data_analysis/tool.py
View File

@ -5,7 +5,7 @@ import json
import os import os
from io import StringIO from io import StringIO
from sys import version_info from sys import version_info
from typing import IO, TYPE_CHECKING, Any, Callable, List, Optional, Type from typing import IO, TYPE_CHECKING, Any, Callable, List, Optional, Type, Union
from langchain_core.callbacks import ( from langchain_core.callbacks import (
AsyncCallbackManagerForToolRun, AsyncCallbackManagerForToolRun,
@ -197,11 +197,11 @@ class E2BDataAnalysisTool(BaseTool):
"exit_code": output.exit_code, "exit_code": output.exit_code,
} }
def install_python_packages(self, package_names: str | List[str]) -> None: def install_python_packages(self, package_names: Union[str, List[str]]) -> None:
"""Install python packages in the sandbox.""" """Install python packages in the sandbox."""
self.session.install_python_packages(package_names) self.session.install_python_packages(package_names)
def install_system_packages(self, package_names: str | List[str]) -> None: def install_system_packages(self, package_names: Union[str, List[str]]) -> None:
"""Install system packages (via apt) in the sandbox.""" """Install system packages (via apt) in the sandbox."""
self.session.install_system_packages(package_names) self.session.install_system_packages(package_names)

4
libs/core/langchain_core/output_parsers/base.py
View File

@ -96,7 +96,7 @@ class BaseGenerationOutputParser(
async def ainvoke( async def ainvoke(
self, self,
input: str | BaseMessage, input: Union[str, BaseMessage],
config: Optional[RunnableConfig] = None, config: Optional[RunnableConfig] = None,
**kwargs: Optional[Any], **kwargs: Optional[Any],
) -> T: ) -> T:
@ -185,7 +185,7 @@ class BaseOutputParser(
async def ainvoke( async def ainvoke(
self, self,
input: str | BaseMessage, input: Union[str, BaseMessage],
config: Optional[RunnableConfig] = None, config: Optional[RunnableConfig] = None,
**kwargs: Optional[Any], **kwargs: Optional[Any],
) -> T: ) -> T:

2
libs/core/langchain_core/tracers/base.py
View File

@ -118,7 +118,7 @@ class BaseTracer(BaseCallbackHandler, ABC):
return parent_run.child_execution_order + 1 return parent_run.child_execution_order + 1
def _get_run(self, run_id: UUID, run_type: str | None = None) -> Run: def _get_run(self, run_id: UUID, run_type: Optional[str] = None) -> Run:
try: try:
run = self.run_map[str(run_id)] run = self.run_map[str(run_id)]
except KeyError as exc: except KeyError as exc: