github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-03-18 19:18:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix case for empty lists

Browse Source
This commit is contained in:
Ankush Gola
2023-06-06 22:55:34 -07:00
parent 5fa7ea2146
commit 40409b4dbf
2 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
langchain/chat_models/base.py
View File

@@ -82,7 +82,9 @@ class BaseChatModel(BaseLanguageModel, ABC):
)
try:
results = [
self._generate(m, stop=stop, run_manager=run_managers[0])
self._generate(
m, stop=stop, run_manager=run_managers[0] if run_managers else None
)
if new_arg_supported
else self._generate(m, stop=stop)
for m in messages
@@ -126,7 +128,11 @@ class BaseChatModel(BaseLanguageModel, ABC):
try:
results = await asyncio.gather(
*[
self._agenerate(m, stop=stop, run_manager=run_managers[0])
self._agenerate(
m,
stop=stop,
run_manager=run_managers[0] if run_managers else None,
)
if new_arg_supported
else self._agenerate(m, stop=stop)
for m in messages

10
langchain/llms/base.py
View File

@@ -184,7 +184,11 @@ class BaseLLM(BaseLanguageModel, ABC):
)
try:
output = (
self._generate(prompts, stop=stop, run_manager=run_managers[0])
self._generate(
prompts,
stop=stop,
run_manager=run_managers[0] if run_managers else None,
)
if new_arg_supported
else self._generate(prompts, stop=stop)
)
@@ -209,7 +213,9 @@ class BaseLLM(BaseLanguageModel, ABC):
try:
new_results = (
self._generate(
missing_prompts, stop=stop, run_manager=run_managers[0]
missing_prompts,
stop=stop,
run_manager=run_managers[0] if run_managers else None,
)
if new_arg_supported
else self._generate(missing_prompts, stop=stop)