Merge pull request #48 from nahcol10/main
update the llm wrappers example for google ai / gemini
This commit is contained in:
commit
8418175881
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue