mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-06 05:08:20 +00:00
This commit is contained in:
parent
6a5a2fb9c8
commit
6fb3cc6f27
@ -233,8 +233,8 @@ class GPTRouter(BaseChatModel):
|
||||
self,
|
||||
messages: List[BaseMessage],
|
||||
stop: Optional[List[str]] = None,
|
||||
run_manager: CallbackManagerForLLMRun | None = None,
|
||||
stream: bool | None = None,
|
||||
run_manager: Optional[CallbackManagerForLLMRun] = None,
|
||||
stream: Optional[bool] = None,
|
||||
**kwargs: Any,
|
||||
) -> ChatResult:
|
||||
should_stream = stream if stream is not None else self.streaming
|
||||
@ -259,8 +259,8 @@ class GPTRouter(BaseChatModel):
|
||||
self,
|
||||
messages: List[BaseMessage],
|
||||
stop: Optional[List[str]] = None,
|
||||
run_manager: AsyncCallbackManagerForLLMRun | None = None,
|
||||
stream: bool | None = None,
|
||||
run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
|
||||
stream: Optional[bool] = None,
|
||||
**kwargs: Any,
|
||||
) -> ChatResult:
|
||||
should_stream = stream if stream is not None else self.streaming
|
||||
|
@ -5,7 +5,7 @@ import json
|
||||
import os
|
||||
from io import StringIO
|
||||
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 (
|
||||
AsyncCallbackManagerForToolRun,
|
||||
@ -197,11 +197,11 @@ class E2BDataAnalysisTool(BaseTool):
|
||||
"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."""
|
||||
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."""
|
||||
self.session.install_system_packages(package_names)
|
||||
|
||||
|
@ -96,7 +96,7 @@ class BaseGenerationOutputParser(
|
||||
|
||||
async def ainvoke(
|
||||
self,
|
||||
input: str | BaseMessage,
|
||||
input: Union[str, BaseMessage],
|
||||
config: Optional[RunnableConfig] = None,
|
||||
**kwargs: Optional[Any],
|
||||
) -> T:
|
||||
@ -185,7 +185,7 @@ class BaseOutputParser(
|
||||
|
||||
async def ainvoke(
|
||||
self,
|
||||
input: str | BaseMessage,
|
||||
input: Union[str, BaseMessage],
|
||||
config: Optional[RunnableConfig] = None,
|
||||
**kwargs: Optional[Any],
|
||||
) -> T:
|
||||
|
@ -118,7 +118,7 @@ class BaseTracer(BaseCallbackHandler, ABC):
|
||||
|
||||
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:
|
||||
run = self.run_map[str(run_id)]
|
||||
except KeyError as exc:
|
||||
|
Loading…
Reference in New Issue
Block a user