Skip to content

Commit 62e6b2a

Browse files
fix: import
1 parent cbb1514 commit 62e6b2a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apps/backend/src/rhesis/backend/app/services/gemini_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from google import genai
1010
from google.genai import errors
1111

12-
from rhesis.backend.app.constants import DEFAULT_MODEL_NAME
12+
from rhesis.backend.app.constants import DEFAULT_LANGUAGE_MODEL_NAME
1313

1414
# Configure logging
1515
logging.basicConfig(
@@ -35,7 +35,7 @@ def get_client():
3535

3636
def _get_model_name(client):
3737
"""Get the appropriate model name based on environment variables."""
38-
return os.getenv("GEMINI_MODEL_NAME", DEFAULT_MODEL_NAME)
38+
return os.getenv("GEMINI_MODEL_NAME", DEFAULT_LANGUAGE_MODEL_NAME)
3939

4040

4141
def _with_retries(func, *args, **kwargs):

0 commit comments

Comments
 (0)