Skip to content

Commit 76ce257

Browse files
committed
fix: update to pydantic documentation
1 parent d55f6be commit 76ce257

19 files changed

+48
-21
lines changed

examples/anthropic/code_generator_graph_anthropic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/azure/code_generator_graph_azure.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/bedrock/code_generator_graph_bedrock.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/deepseek/code_generator_graph_deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/ernie/code_generator_graph_ernie.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/fireworks/code_generator_graph_fireworks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/google_genai/code_generator_graph_gemini.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/google_vertexai/code_generator_graph_vertex.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/groq/code_generator_graph_groq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/huggingfacehub/code_generator_graph_huggingfacehub.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010
from langchain_community.llms import HuggingFaceEndpoint
1111
from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings

examples/local_models/code_generator_graph_ollama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/mistral/code_generator_graph_mistral.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/moonshot/code_generator_graph_moonshot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from langchain_community.chat_models.moonshot import MoonshotChat
1010
from scrapegraphai.graphs import CodeGeneratorGraph
1111

examples/nemotron/code_generator_graph_nemotron.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/oneapi/code_generator_graph_oneapi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

examples/openai/code_generator_graph_openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import os, json
66
from typing import List
77
from dotenv import load_dotenv
8-
from langchain_core.pydantic_v1 import BaseModel, Field
8+
from pydantic import BaseModel, Field
99
from scrapegraphai.graphs import CodeGeneratorGraph
1010

1111
load_dotenv()

extracted_data.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
def extract_data(html: str) -> dict:
2+
from bs4 import BeautifulSoup
3+
4+
# Parse the HTML content using BeautifulSoup
5+
soup = BeautifulSoup(html, 'html.parser')
6+
7+
# Initialize an empty list to hold project data
8+
projects = []
9+
10+
# Find all project entries in the HTML
11+
project_entries = soup.find_all('div', class_='grid-item')
12+
13+
# Iterate over each project entry to extract title and description
14+
for entry in project_entries:
15+
# Extract the title from the card-title class
16+
title = entry.find('h4', class_='card-title').get_text(strip=True)
17+
18+
# Extract the description from the card-text class
19+
description = entry.find('p', class_='card-text').get_text(strip=True)
20+
21+
# Append the extracted data as a dictionary to the projects list
22+
projects.append({
23+
'title': title,
24+
'description': description
25+
})
26+
27+
# Return the structured data as a dictionary matching the desired JSON schema
28+
return {'projects': projects}

scrapegraphai/nodes/fetch_node.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -319,4 +319,3 @@ def handle_web_source(self, state, source):
319319
state["original_html"] = document
320320
state.update({self.output[0]: compressed_document,})
321321
return state
322-

scrapegraphai/utils/schema_trasform.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,15 @@ def transform_schema(pydantic_schema):
1212
Returns:
1313
dict: The transformed JSON schema.
1414
"""
15-
15+
1616
def process_properties(properties):
1717
result = {}
1818
for key, value in properties.items():
1919
if 'type' in value:
2020
if value['type'] == 'array':
2121
if '$ref' in value['items']:
2222
ref_key = value['items']['$ref'].split('/')[-1]
23-
result[key] = [process_properties(pydantic_schema['definitions'][ref_key]['properties'])]
23+
result[key] = [process_properties(pydantic_schema['$defs'][ref_key]['properties'])]
2424
else:
2525
result[key] = [value['items']['type']]
2626
else:
@@ -30,7 +30,7 @@ def process_properties(properties):
3030
}
3131
elif '$ref' in value:
3232
ref_key = value['$ref'].split('/')[-1]
33-
result[key] = process_properties(pydantic_schema['definitions'][ref_key]['properties'])
33+
result[key] = process_properties(pydantic_schema['$defs'][ref_key]['properties'])
3434
return result
3535

36-
return process_properties(pydantic_schema['properties'])
36+
return process_properties(pydantic_schema['properties'])

0 commit comments

Comments
 (0)