update the llm wrappers example for google ai / gemini
This commit is contained in:
@@ -45,13 +45,13 @@ Here, we provide some minimal example implementations:
|
||||
3. Google (Generative AI Studio / PaLM API)
|
||||
```python
|
||||
def call_llm(prompt):
|
||||
import google.generativeai as genai
|
||||
genai.configure(api_key="YOUR_API_KEY_HERE")
|
||||
r = genai.generate_text(
|
||||
model="models/text-bison-001",
|
||||
prompt=prompt
|
||||
from google import genai
|
||||
client = genai.Client(api_key='GEMINI_API_KEY')
|
||||
response = client.models.generate_content(
|
||||
model='gemini-2.0-flash-001',
|
||||
contents=prompt
|
||||
)
|
||||
return r.result
|
||||
return response.text
|
||||
```
|
||||
|
||||
4. Azure (Azure OpenAI)
|
||||
|
||||
Reference in New Issue
Block a user