CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update generate_answer_from_image_node.py
Some checks are pending
/ build (push) Waiting to run
Details

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-08-19 02:21:45 +02:00
parent 79fa3f6bd6
commit 0bf79b5926
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/nodes/generate_answer_from_image_node.py
View File

@ -32,8 +32,6 @@ class GenerateAnswerFromImageNode(BaseNode):
images = state.get('screenshots', [])
analyses = []
api_key = self.node_config.get("config", {}).get("llm", {}).get("api_key", "")
supported_models = ("gpt-4o", "gpt-4o-mini", "gpt-4-turbo")
if self.node_config["config"]["llm"]["model"] not in supported_models:
@ -42,6 +40,8 @@ class GenerateAnswerFromImageNode(BaseNode):
{', '.join(supported_models)}.""")
if self.node_config["config"]["llm"]["model"].startswith("gpt"):
api_key = self.node_config.get("config", {}).get("llm", {}).get("api_key", "")
for image_data in images:
base64_image = base64.b64encode(image_data).decode('utf-8')