Fix vertex ai response parsing bug.

eli64s · Mar 1, 2024 · 3afc418 · 3afc418
1 parent 45068eb
commit 3afc418
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "readmeai"
-version = "0.5.069"
+version = "0.5.070"
 description = "👾 Automated README file generator, powered by LLM APIs."
 authors = ["Eli <egsalamie@gmail.com>"]
 license = "MIT"

diff --git a/readmeai/models/vertex.py b/readmeai/models/vertex.py
@@ -76,7 +76,7 @@ async def _make_request(
                     prompt,
                     generation_config=data,
                 )
-                content = response.text if response.text else response
+                content = response.candidates[0].content.parts[0].text
                 self._logger.info(f"Response for '{index}':\n{content}")
                 return index, clean_response(index, content)