Lhumpal commited on
Commit
c107fb9
·
verified ·
1 Parent(s): 015794e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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": "I have 10 dogs"}]},
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(