Update app.py
Browse files
app.py
CHANGED
@@ -46,6 +46,8 @@ async def chat(request: ChatRequest):
|
|
46 |
return {"response": response.choices[0].message.content}
|
47 |
|
48 |
if request.model_choice == "google":
|
|
|
|
|
49 |
messages = [
|
50 |
{"role": "user", "parts": [{"text": "I have 10 dogs"}]},
|
51 |
{"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
|
|
|
46 |
return {"response": response.choices[0].message.content}
|
47 |
|
48 |
if request.model_choice == "google":
|
49 |
+
client = genai.Client(api_key=google_api_key)
|
50 |
+
|
51 |
messages = [
|
52 |
{"role": "user", "parts": [{"text": "I have 10 dogs"}]},
|
53 |
{"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
|