Skip to content

Commit f6df9b7

Browse files
f-aguzzif-aguzzi
committed
chore(examples): update model names
1 parent f73343f commit f6df9b7

File tree

100 files changed

+192
-257
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

100 files changed

+192
-257
lines changed

‎examples/anthropic/csv_scraper_graph_multi_haiku.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
graph_config = {
2727
"llm": {
2828
"api_key": os.getenv("ANTHROPIC_API_KEY"),
29-
"model": "claude-3-haiku-20240307",
30-
"max_tokens": 4000},
29+
"model": "anthropic/claude-3-haiku-20240307",
30+
},
3131
}
3232

3333
# ************************************************

‎examples/anthropic/csv_scraper_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,8 @@
3232
graph_config = {
3333
"llm": {
3434
"api_key": os.getenv("ANTHROPIC_API_KEY"),
35-
"model": "claude-3-haiku-20240307",
36-
"max_tokens": 4000
37-
},
35+
"model": "anthropic/claude-3-haiku-20240307",
36+
},
3837
}
3938

4039
# ************************************************

‎examples/anthropic/custom_graph_haiku.py

Lines changed: 5 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,9 @@
55
import os
66
from dotenv import load_dotenv
77

8-
from langchain_openai import OpenAIEmbeddings
9-
from langchain_openai import ChatOpenAI
8+
from langchain_anthropic import ChatAnthropic
109
from scrapegraphai.graphs import BaseGraph
11-
from scrapegraphai.nodes import FetchNode, ParseNode, RAGNode, GenerateAnswerNode, RobotsNode
10+
from scrapegraphai.nodes import FetchNode, ParseNode, GenerateAnswerNode, RobotsNode
1211
load_dotenv()
1312

1413
# ************************************************
@@ -19,16 +18,14 @@
1918
"llm": {
2019
"api_key": os.getenv("ANTHROPIC_API_KEY"),
2120
"model": "claude-3-haiku-20240307",
22-
"max_tokens": 4000
23-
},
21+
},
2422
}
2523

2624
# ************************************************
2725
# Define the graph nodes
2826
# ************************************************
2927

30-
llm_model = OpenAI(graph_config["llm"])
31-
embedder = OpenAIEmbeddings(api_key=llm_model.openai_api_key)
28+
llm_model = ChatAnthropic(graph_config["llm"])
3229

3330
# define the nodes for the graph
3431
robot_node = RobotsNode(
@@ -57,15 +54,6 @@
5754
"verbose": True,
5855
}
5956
)
60-
rag_node = RAGNode(
61-
input="user_prompt & (parsed_doc | doc)",
62-
output=["relevant_chunks"],
63-
node_config={
64-
"llm_model": llm_model,
65-
"embedder_model": embedder,
66-
"verbose": True,
67-
}
68-
)
6957
generate_answer_node = GenerateAnswerNode(
7058
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
7159
output=["answer"],
@@ -84,14 +72,12 @@
8472
robot_node,
8573
fetch_node,
8674
parse_node,
87-
rag_node,
8875
generate_answer_node,
8976
],
9077
edges=[
9178
(robot_node, fetch_node),
9279
(fetch_node, parse_node),
93-
(parse_node, rag_node),
94-
(rag_node, generate_answer_node)
80+
(parse_node, generate_answer_node)
9581
],
9682
entry_point=robot_node
9783
)

‎examples/anthropic/json_scraper_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,8 @@
2626
graph_config = {
2727
"llm": {
2828
"api_key": os.getenv("ANTHROPIC_API_KEY"),
29-
"model": "claude-3-haiku-20240307",
30-
"max_tokens": 4000
31-
},
29+
"model": "anthropic/claude-3-haiku-20240307",
30+
},
3231
}
3332

3433
# ************************************************

‎examples/anthropic/json_scraper_multi_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,8 @@
1111
graph_config = {
1212
"llm": {
1313
"api_key": os.getenv("ANTHROPIC_API_KEY"),
14-
"model": "claude-3-haiku-20240307",
15-
"max_tokens": 4000
16-
},
14+
"model": "anthropic/claude-3-haiku-20240307",
15+
},
1716
}
1817

1918
FILE_NAME = "inputs/example.json"

‎examples/anthropic/pdf_scraper_graph_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,8 @@
1414
graph_config = {
1515
"llm": {
1616
"api_key": os.getenv("ANTHROPIC_API_KEY"),
17-
"model": "claude-3-haiku-20240307",
18-
"max_tokens": 4000
19-
},
17+
"model": "anthropic/claude-3-haiku-20240307",
18+
},
2019
}
2120

2221
source = """

‎examples/anthropic/pdf_scraper_multi_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,8 @@
1111
graph_config = {
1212
"llm": {
1313
"api_key": os.getenv("ANTHROPIC_API_KEY"),
14-
"model": "claude-3-haiku-20240307",
15-
"max_tokens": 4000
16-
},
14+
"model": "anthropic/claude-3-haiku-20240307",
15+
},
1716
}
1817

1918
# ***************

‎examples/anthropic/scrape_plain_text_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,8 @@
2828
graph_config = {
2929
"llm": {
3030
"api_key": os.getenv("ANTHROPIC_API_KEY"),
31-
"model": "claude-3-haiku-20240307",
32-
"max_tokens": 4000
33-
},
31+
"model": "anthropic/claude-3-haiku-20240307",
32+
},
3433
}
3534

3635
# ************************************************

‎examples/anthropic/script_generator_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,8 @@
1616
graph_config = {
1717
"llm": {
1818
"api_key": os.getenv("ANTHROPIC_API_KEY"),
19-
"model": "claude-3-haiku-20240307",
20-
"max_tokens": 4000
21-
},
19+
"model": "anthropic/claude-3-haiku-20240307",
20+
},
2221
}
2322

2423
# ************************************************

‎examples/anthropic/script_multi_generator_haiku.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,9 @@
1616
graph_config = {
1717
"llm": {
1818
"api_key": os.getenv("ANTHROPIC_API_KEY"),
19-
"model": "claude-3-haiku-20240307",
20-
"max_tokens": 4000
21-
},
22-
"library": "beautifulsoup"
19+
"model": "anthropic/claude-3-haiku-20240307",
20+
},
21+
"library": "beautifulsoup"
2322
}
2423

2524
# ************************************************

‎examples/anthropic/search_graph_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,8 @@
1515
graph_config = {
1616
"llm": {
1717
"api_key": os.getenv("ANTHROPIC_API_KEY"),
18-
"model": "claude-3-haiku-20240307",
19-
"max_tokens": 4000
20-
},
18+
"model": "anthropic/claude-3-haiku-20240307",
19+
},
2120
}
2221

2322
# ************************************************

‎examples/anthropic/search_graph_schema_haiku.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@ class Dishes(BaseModel):
2727
graph_config = {
2828
"llm": {
2929
"api_key": os.getenv("ANTHROPIC_API_KEY"),
30-
"model": "claude-3-haiku-20240307",
31-
"max_tokens": 4000},
30+
"model": "anthropic/claude-3-haiku-20240307",
31+
},
3232
}
3333

3434
# ************************************************

‎examples/anthropic/search_link_graph_haiku.py

Lines changed: 4 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -14,23 +14,11 @@
1414

1515
load_dotenv()
1616

17-
llm_model_instance = AzureChatOpenAI(
18-
openai_api_version=os.environ["AZURE_OPENAI_API_VERSION"],
19-
azure_deployment=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT_NAME"]
20-
)
21-
22-
embedder_model_instance = AzureOpenAIEmbeddings(
23-
azure_deployment=os.environ["AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT_NAME"],
24-
openai_api_version=os.environ["AZURE_OPENAI_API_VERSION"],
25-
)
26-
27-
# ************************************************
28-
# Create the SmartScraperGraph instance and run it
29-
# ************************************************
30-
3117
graph_config = {
32-
"llm": {"model_instance": llm_model_instance},
33-
"embeddings": {"model_instance": embedder_model_instance}
18+
"llm": {
19+
"api_key": os.getenv("ANTHROPIC_API_KEY"),
20+
"model": "anthropic/claude-3-haiku-20240307",
21+
},
3422
}
3523

3624
# ************************************************

‎examples/anthropic/smart_scraper_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,8 @@
1919
graph_config = {
2020
"llm": {
2121
"api_key": os.getenv("ANTHROPIC_API_KEY"),
22-
"model": "claude-3-haiku-20240307",
23-
"max_tokens": 4000
24-
},
22+
"model": "anthropic/claude-3-haiku-20240307",
23+
},
2524
}
2625

2726
smart_scraper_graph = SmartScraperGraph(

‎examples/anthropic/smart_scraper_multi_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,8 @@
1717
graph_config = {
1818
"llm": {
1919
"api_key": os.getenv("ANTHROPIC_API_KEY"),
20-
"model": "claude-3-haiku-20240307",
21-
"max_tokens": 4000
22-
},
20+
"model": "anthropic/claude-3-haiku-20240307",
21+
},
2322
}
2423

2524
# *******************************************************

‎examples/anthropic/smart_scraper_schema_haiku.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ class Projects(BaseModel):
3333
graph_config = {
3434
"llm": {
3535
"api_key": os.getenv("ANTHROPIC_API_KEY"),
36-
"model": "claude-3-haiku-20240307",
37-
"max_tokens": 4000},
36+
"model": "anthropic/claude-3-haiku-20240307",
37+
},
3838
}
3939

4040
smart_scraper_graph = SmartScraperGraph(

‎examples/anthropic/xml_scraper_graph_multi_haiku.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
graph_config = {
2727
"llm": {
2828
"api_key": os.getenv("ANTHROPIC_API_KEY"),
29-
"model": "claude-3-haiku-20240307",
30-
"max_tokens": 4000},
29+
"model": "anthropic/claude-3-haiku-20240307",
30+
},
3131
}
3232

3333
# ************************************************

‎examples/anthropic/xml_scraper_haiku.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,8 @@
2626
graph_config = {
2727
"llm": {
2828
"api_key": os.getenv("ANTHROPIC_API_KEY"),
29-
"model": "claude-3-haiku-20240307",
30-
"max_tokens": 4000
31-
},
29+
"model": "anthropic/claude-3-haiku-20240307",
30+
},
3231
}
3332

3433
# ************************************************

‎examples/deepseek/csv_scraper_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
graph_config = {
2929
"llm": {
30-
"model": "deepseek-chat",
30+
"model": "deepseek/deepseek-chat",
3131
"openai_api_key": deepseek_key,
3232
"openai_api_base": 'https://api.deepseek.com/v1',
3333
},

‎examples/deepseek/csv_scraper_graph_multi_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
graph_config = {
2929
"llm": {
30-
"model": "deepseek-chat",
30+
"model": "deepseek/deepseek-chat",
3131
"openai_api_key": deepseek_key,
3232
"openai_api_base": 'https://api.deepseek.com/v1',
3333
},

‎examples/deepseek/json_scraper_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626

2727
graph_config = {
2828
"llm": {
29-
"model": "deepseek-chat",
29+
"model": "deepseek/deepseek-chat",
3030
"openai_api_key": deepseek_key,
3131
"openai_api_base": 'https://api.deepseek.com/v1',
3232
},

‎examples/deepseek/json_scraper_multi_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
graph_config = {
1414
"llm": {
15-
"model": "deepseek-chat",
15+
"model": "deepseek/deepseek-chat",
1616
"openai_api_key": deepseek_key,
1717
"openai_api_base": 'https://api.deepseek.com/v1',
1818
},

‎examples/deepseek/pdf_scraper_graph_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
graph_config = {
1919
"llm": {
20-
"model": "deepseek-chat",
20+
"model": "deepseek/deepseek-chat",
2121
"openai_api_key": deepseek_key,
2222
"openai_api_base": 'https://api.deepseek.com/v1',
2323
},

‎examples/deepseek/pdf_scraper_multi_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
graph_config = {
1414
"llm": {
15-
"model": "deepseek-chat",
15+
"model": "deepseek/deepseek-chat",
1616
"openai_api_key": deepseek_key,
1717
"openai_api_base": 'https://api.deepseek.com/v1',
1818
},

‎examples/deepseek/scrape_plain_text_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929
graph_config = {
3030
"llm": {
31-
"model": "deepseek-chat",
31+
"model": "deepseek/deepseek-chat",
3232
"openai_api_key": deepseek_key,
3333
"openai_api_base": 'https://api.deepseek.com/v1',
3434
},

‎examples/deepseek/script_generator_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
graph_config = {
1919
"llm": {
20-
"model": "deepseek-chat",
20+
"model": "deepseek/deepseek-chat",
2121
"openai_api_key": deepseek_key,
2222
"openai_api_base": 'https://api.deepseek.com/v1',
2323
},

‎examples/deepseek/script_multi_generator_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
graph_config = {
1919
"llm": {
20-
"model": "deepseek-chat",
20+
"model": "deepseek/deepseek-chat",
2121
"openai_api_key": deepseek_key,
2222
"openai_api_base": 'https://api.deepseek.com/v1',
2323
},

‎examples/deepseek/search_graph_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616
graph_config = {
1717
"llm": {
18-
"model": "deepseek-chat",
18+
"model": "deepseek/deepseek-chat",
1919
"openai_api_key": deepseek_key,
2020
"openai_api_base": 'https://api.deepseek.com/v1',
2121
},

‎examples/deepseek/search_graph_schema_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class Dishes(BaseModel):
3131

3232
graph_config = {
3333
"llm": {
34-
"model": "deepseek-chat",
34+
"model": "deepseek/deepseek-chat",
3535
"openai_api_key": deepseek_key,
3636
"openai_api_base": 'https://api.deepseek.com/v1',
3737
},

‎examples/deepseek/search_link_graph_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
graph_config = {
1818
"llm": {
19-
"model": "deepseek-chat",
19+
"model": "deepseek/deepseek-chat",
2020
"openai_api_key": deepseek_key,
2121
"openai_api_base": 'https://api.deepseek.com/v1',
2222
},

‎examples/deepseek/smart_scraper_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
graph_config = {
2020
"llm": {
21-
"model": "deepseek-chat",
21+
"model": "deepseek/deepseek-chat",
2222
"openai_api_key": deepseek_key,
2323
"openai_api_base": 'https://api.deepseek.com/v1',
2424
},

0 commit comments

Comments
 (0)