Skip to content

Commit 3ae2ea1

Browse files
committed
Miscellaneous "llm" -> "llm_model" refactors
1 parent a53e95c commit 3ae2ea1

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

examples/openai/custom_graph_openai.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
robot_node = RobotsNode(
3535
input="url",
3636
output=["is_scrapable"],
37-
node_config={"llm": llm_model}
37+
node_config={"llm_model": llm_model}
3838
)
3939

4040
fetch_node = FetchNode(
@@ -50,12 +50,12 @@
5050
rag_node = RAGNode(
5151
input="user_prompt & (parsed_doc | doc)",
5252
output=["relevant_chunks"],
53-
node_config={"llm": llm_model},
53+
node_config={"llm_model": llm_model},
5454
)
5555
generate_answer_node = GenerateAnswerNode(
5656
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
5757
output=["answer"],
58-
node_config={"llm": llm_model},
58+
node_config={"llm_model": llm_model},
5959
)
6060

6161
# ************************************************

scrapegraphai/graphs/json_scraper_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def _create_graph(self) -> BaseGraph:
7676
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
7777
output=["answer"],
7878
node_config={
79-
"llm": self.llm_model
79+
"llm_model": self.llm_model
8080
}
8181
)
8282

scrapegraphai/nodes/fetch_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class FetchNode(BaseNode):
2929
node_name (str): The unique identifier name for the node, defaulting to "Fetch".
3030
"""
3131

32-
def __init__(self, input: str, output: List[str], node_config: Optional[dict], node_name: str = "Fetch"):
32+
def __init__(self, input: str, output: List[str], node_config: Optional[dict]=None, node_name: str = "Fetch"):
3333
super().__init__(node_name, "node", input, output, 1)
3434

3535
self.headless = True if node_config is None else node_config.get("headless", True)

0 commit comments

Comments
 (0)