Update app.py
Browse files
app.py
CHANGED
@@ -50,9 +50,9 @@ async def chat(request: ChatRequest):
|
|
50 |
client = genai.Client(api_key=google_api_key)
|
51 |
|
52 |
messages = [
|
53 |
-
{"role": "user", "parts": [{"text":
|
54 |
-
{"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
|
55 |
-
{"role": "user", "parts": [{"text": "How many dogs do I have?"}]},
|
56 |
]
|
57 |
|
58 |
response = client.models.generate_content(
|
|
|
50 |
client = genai.Client(api_key=google_api_key)
|
51 |
|
52 |
messages = [
|
53 |
+
{"role": "user", "parts": [{"text": request.message}]},
|
54 |
+
# {"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
|
55 |
+
# {"role": "user", "parts": [{"text": "How many dogs do I have?"}]},
|
56 |
]
|
57 |
|
58 |
response = client.models.generate_content(
|