Skip to content

Commit d332e21

Browse files
committed
fix: fix generate answer node
1 parent 6f1a88f commit d332e21

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

scrapegraphai/nodes/generate_answer_node.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from langchain_core.runnables import RunnableParallel
99
from langchain_openai import ChatOpenAI, AzureChatOpenAI
1010
from langchain_aws import ChatBedrock
11-
from langchain_mistralai import ChatMistralAI
1211
from langchain_community.chat_models import ChatOllama
1312
from tqdm import tqdm
1413
from .base_node import BaseNode
@@ -77,7 +76,7 @@ def execute(self, state: dict) -> dict:
7776
doc = input_data[1]
7877

7978
if self.node_config.get("schema", None) is not None:
80-
if isinstance(self.llm_model, (ChatOpenAI, ChatMistralAI)):
79+
if isinstance(self.llm_model, ChatOpenAI):
8180
self.llm_model = self.llm_model.with_structured_output(
8281
schema=self.node_config["schema"]
8382
)

0 commit comments

Comments
 (0)