add fix
This commit is contained in:
parent
f2a00f35c4
commit
c255685a4f
@ -6,6 +6,6 @@ APP__POSTGRES__HOST=postgres
|
||||
APP__POSTGRES__PORT=5432
|
||||
APP__POSTGRES__PASSWORD=postgres
|
||||
|
||||
APP__RABBIT__URL=amqp://test:test@localhost:9072/
|
||||
APP__RABBIT__URL=amqp://test:test@103.88.243.161:9072/
|
||||
|
||||
APP__GEMINI__API_KEY=AIzaSyCxGFlj221gTBSMGoftD4NjMioU88jI2cQ
|
1
gemini_worker/.gitignore
vendored
Normal file
1
gemini_worker/.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
||||
.env_prod
|
@ -55,12 +55,9 @@ class GoogleHelper:
|
||||
messages: MessagesForSendToWorkersSchema,
|
||||
) -> ResponseFromGeminiSchema:
|
||||
contents = self._serialize_messages_to_promt(messages)
|
||||
for msg in messages.messages:
|
||||
print(msg.text)
|
||||
response = self._model.generate_content(
|
||||
contents=contents
|
||||
)
|
||||
print(response)
|
||||
|
||||
return self._serialize_response_to_json(response.text)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user