Fixes #166. Thanks @piresramon. (#208)

Author: MarkDaoust

google/generativeai/types/content_types.py

@@ -93,6 +93,8 @@ class BlobDict(TypedDict):
 def _convert_dict(d: Mapping) -> glm.Content | glm.Part | glm.Blob:
     if is_content_dict(d):
         content = dict(d)
+        if isinstance(parts := content["parts"], str):
+            content["parts"] = [parts]
         content["parts"] = [to_part(part) for part in content["parts"]]
         return glm.Content(content)
     elif is_part_dict(d):

tests/test_content.py

@@ -137,12 +137,14 @@ def test_strict_to_contents_fails(self, examples):
         ["glm.Content", [glm.Content(parts=[{"text": "Hello world!"}])]],
         ["ContentDict", [{"parts": [{"text": "Hello world!"}]}]],
         ["ContentDict-unwraped", [{"parts": ["Hello world!"]}]],
+        ["ContentDict+str-part", [{"parts": "Hello world!"}]],
     )
     def test_to_contents(self, example):
         contents = content_types.to_contents(example)
         part = contents[0].parts[0]
 
         self.assertLen(contents, 1)
+
         self.assertLen(contents[0].parts, 1)
         self.assertIsInstance(part, glm.Part)
         self.assertEqual(part.text, "Hello world!")