Skip to content

Commit 2d842dc

Browse files
committed
try fix max_retries from google call
1 parent 04ad3c6 commit 2d842dc

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/agents/llm_factory.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ def __init__(self, temperature: int = 0):
7979
model=os.environ['GOOGLE_AI_MODEL'],
8080
temperature=temperature,
8181
google_api_key=os.environ['GOOGLE_AI_API_KEY'],
82+
max_retries=0,
8283
)
8384

8485
def get_llm(self):

src/module.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import time
22
from typing import Any
3-
from lf_toolkit.chat import ChatResult as Result, ChatParams as Params
3+
from lf_toolkit.chat.result import ChatResult as Result
4+
from lf_toolkit.chat.params import ChatParams as Params
45

56
try:
67
from .agents.utils.parse_json_context_to_prompt import parse_json_to_prompt

0 commit comments

Comments
 (0)