Skip to content

Commit 3b4e9fb

Browse files
authored
Refactor async chat code (#47)
1 parent 8abd8f7 commit 3b4e9fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

google/generativeai/discuss.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ def chat(
346346
@set_doc(chat.__doc__)
347347
async def chat_async(
348348
*,
349-
model: model_types.ModelNameOptions | None = None,
349+
model: model_types.ModelNameOptions | None = "models/chat-bison-001",
350350
context: str | None = None,
351351
examples: discuss_types.ExamplesOptions | None = None,
352352
messages: discuss_types.MessagesOptions | None = None,
@@ -436,7 +436,7 @@ async def reply_async(
436436
)
437437
request = self.to_dict()
438438
request.pop("candidates")
439-
request.pop("filters")
439+
request.pop("filters", None)
440440
request["messages"] = list(request["messages"])
441441
request["messages"].append(_make_message(message))
442442
request = _make_generate_message_request(**request)

0 commit comments

Comments
 (0)