running taskwaver via python notebook showing error on gemini prompt

[yadavj2008]

Describe the bug I'm trying to run the taskwearver on colab notebook by importing taskweaver as python package, i was able to create the session with Gemini model but when I tried running the prompt I faced below error.

To Reproduce

Steps to reproduce the behavior:

1. Start prompt with Gemini session.

user_query = "hello, what can you do?"
response_round = session.send_message(user_query,
                                      event_handler=lambda _type, _msg: print(f"{_type}:\n{_msg}"))

2. See error

INFO:taskweaver.logging:Use back up engine: False
ERROR:taskweaver.logging:Traceback (most recent call last):
  File "/usr/local/lib/python3.10/dist-packages/google/protobuf/json_format.py", line 544, in _ConvertFieldValuePair
    raise ParseError(
google.protobuf.json_format.ParseError: Message type "google.ai.generativelanguage.v1beta.GenerateContentResponse" has no field named "error" at "GenerateContentResponse".
 Available Fields(except extensions): "['candidates', 'promptFeedback']"

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/usr/local/lib/python3.10/dist-packages/google/generativeai/types/generation_types.py", line 345, in rewrite_stream_error
    yield
  File "/usr/local/lib/python3.10/dist-packages/google/generativeai/types/generation_types.py", line 388, in from_iterator
    response = next(iterator)
  File "/usr/local/lib/python3.10/dist-packages/google/api_core/rest_streaming.py", line 106, in __next__
    return self._grab()
  File "/usr/local/lib/python3.10/dist-packages/google/api_core/rest_streaming.py", line 110, in _grab
    return self._response_message_cls.from_json(self._ready_objs.popleft())
  File "/usr/local/lib/python3.10/dist-packages/proto/message.py", line 428, in from_json
    Parse(payload, instance._pb, ignore_unknown_fields=ignore_unknown_fields)
  File "/usr/local/lib/python3.10/dist-packages/google/protobuf/json_format.py", line 436, in Parse
    return ParseDict(js, message, ignore_unknown_fields, descriptor_pool,
  File "/usr/local/lib/python3.10/dist-packages/google/protobuf/json_format.py", line 461, in ParseDict
    parser.ConvertMessage(js_dict, message, '')
  File "/usr/local/lib/python3.10/dist-packages/google/protobuf/json_format.py", line 502, in ConvertMessage
    self._ConvertFieldValuePair(value, message, path)
  File "/usr/local/lib/python3.10/dist-packages/google/protobuf/json_format.py", line 629, in _ConvertFieldValuePair
    raise ParseError(str(e))
google.protobuf.json_format.ParseError: Message type "google.ai.generativelanguage.v1beta.GenerateContentResponse" has no field named "error" at "GenerateContentResponse".
 Available Fields(except extensions): "['candidates', 'promptFeedback']"

Expected behavior The prompt should run just fine and able to parse the gemini response.

Environment Information (please complete the following information):

• OS: MacOS
• Python Version: 3.10
• LLM that you're using: Gemini pro

Additional context