Skip to content

Commit 9aa889b

Browse files
committed
revert testing changes
1 parent 217977e commit 9aa889b

File tree

3 files changed

+4
-11
lines changed

3 files changed

+4
-11
lines changed

.github/workflows/dev.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ jobs:
4545
run: |
4646
pip install --upgrade pip
4747
pip install -r requirements.txt
48-
pip list
4948
5049
- name: Run tests
5150
if: always()

requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@ langchain
22
langchain-chroma
33
langchain-community
44
langchain-core
5-
langchain-openai==0.3.33
6-
langchain_google_genai==2.1.12
5+
langchain-openai
6+
langchain_google_genai
77
langchain-text-splitters
88
langchainhub
99
langdetect
10-
langgraph==0.6.7
10+
langgraph
1111
langsmith
1212

1313
lf_toolkit[ipc] @ git+https://github.com/lambda-feedback/toolkit-python.git@main

src/agents/llm_factory.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -80,12 +80,6 @@ def __init__(self, temperature: int = 0):
8080
temperature=temperature,
8181
google_api_key=os.environ['GOOGLE_AI_API_KEY'],
8282
)
83-
84-
# This wrapper will catch errors and retry the call itself.
85-
# It does NOT pass 'max_retries' to the underlying model.
86-
self._google_llm_with_retries = self._google_llm.with_retry(
87-
stop_after_attempt=1, # Corresponds to max_retries
88-
)
8983

9084
def get_llm(self):
91-
return self._google_llm_with_retries
85+
return self._google_llm

0 commit comments

Comments
 (0)