We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 374d3eb commit b7a4378Copy full SHA for b7a4378
1 file changed
utils/call_llm.py
@@ -48,6 +48,10 @@ def call_llm(prompt: str, use_cache: bool = True) -> str:
48
project=os.getenv("GEMINI_PROJECT_ID", "your-project-id"),
49
location=os.getenv("GEMINI_LOCATION", "us-central1")
50
)
51
+ # You can comment the previous line and use the AI Studio key instead:
52
+ # client = genai.Client(
53
+ # api_key=os.getenv("GEMINI_API_KEY", "your-api_key"),
54
+ # )
55
model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro-exp-03-25")
56
response = client.models.generate_content(
57
model=model,
@@ -118,4 +122,4 @@ def call_llm(prompt: str, use_cache: bool = True) -> str:
118
122
print("Making call...")
119
123
response1 = call_llm(test_prompt, use_cache=False)
120
124
print(f"Response: {response1}")
121
-
125
+
0 commit comments