Skip to content

554 improve tokenization function #556

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 66 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
66 commits
Select commit Hold shift + click to select a range
1272273
fix: broken node
VinciGit00 Aug 9, 2024
b470d97
ci(release): 1.13.0-beta.8 [skip ci]
semantic-release-bot Aug 9, 2024
6c3b37a
feat: add refactoring of default temperature
VinciGit00 Aug 10, 2024
c3f1520
Update abstract_graph.py
VinciGit00 Aug 10, 2024
a17e068
Merge pull request #529 from ScrapeGraphAI/523-set-default-temperatur…
f-aguzzi Aug 10, 2024
d4c1a1c
ci(release): 1.13.0-beta.9 [skip ci]
semantic-release-bot Aug 10, 2024
8b2c266
refactoring of the code
VinciGit00 Aug 10, 2024
0cda020
Merge branch 'pre/beta' into temp
VinciGit00 Aug 10, 2024
8423f10
Merge pull request #532 from ScrapeGraphAI/temp
VinciGit00 Aug 10, 2024
4ca606c
remove promps form helpers folders
vedovati-matteo Aug 11, 2024
3b5b24d
Added new folder for prompts
vedovati-matteo Aug 11, 2024
9fa3df2
Changed prompts import from helpers to prompts
vedovati-matteo Aug 11, 2024
f17cef9
fix: merge_anwser prompt import
vedovati-matteo Aug 11, 2024
7c62100
Merge branch 'pre/beta' of https://github.com/vedovati-matteo/Scrapeg…
vedovati-matteo Aug 11, 2024
9814b6d
fixed versioning errors
vedovati-matteo Aug 11, 2024
f455fcb
robot_node prompt moved
vedovati-matteo Aug 11, 2024
86a4903
search_internet_node prompt moved
vedovati-matteo Aug 11, 2024
ef96652
search_link_node prompt moved
vedovati-matteo Aug 11, 2024
98779d1
search_link_node prompts moved
vedovati-matteo Aug 11, 2024
734b740
Template name refactoring
vedovati-matteo Aug 11, 2024
a3b7181
quick fix typo
vedovati-matteo Aug 11, 2024
a165fea
Merge pull request #535 from vedovati-matteo/pre/beta
VinciGit00 Aug 11, 2024
40043f3
ci(release): 1.14.0-beta.1 [skip ci]
semantic-release-bot Aug 11, 2024
cec5537
add new tests
VinciGit00 Aug 11, 2024
de1ec25
refactoring pyproject.toml
VinciGit00 Aug 11, 2024
c77231c
feat: update abstract graph
VinciGit00 Aug 11, 2024
cef2fdb
Update abstract_graph.py
VinciGit00 Aug 11, 2024
8cece1d
Merge branch 'pre/beta' into refactoring-of-abstract-graph
VinciGit00 Aug 11, 2024
9e1d0f6
fixing import bug
VinciGit00 Aug 11, 2024
e6bedb6
fix(AbstractGraph): pass kwargs to Ernie and Nvidia models
f-aguzzi Aug 12, 2024
71438a1
chore(examples): fix import bug in image2text demo
f-aguzzi Aug 12, 2024
bae89e8
Merge branch 'pre/beta' into 530-probable-bug
f-aguzzi Aug 12, 2024
7fe181f
chore(requirements): update requirements.txt
f-aguzzi Aug 12, 2024
cb6b353
fix(models_tokens): incorrect provider names
f-aguzzi Aug 12, 2024
8e1c9f2
Merge pull request #537 from ScrapeGraphAI/534-importerror-cannot-imp…
VinciGit00 Aug 12, 2024
3ecd67c
Merge pull request #531 from ScrapeGraphAI/530-probable-bug
f-aguzzi Aug 12, 2024
7fd921b
ci(release): 1.14.0-beta.2 [skip ci]
semantic-release-bot Aug 12, 2024
427666f
fix update Dockerfile
Aug 12, 2024
7378e98
Merge branch 'ScrapeGraphAI:pre/beta' into pre/beta
AmazeQiu Aug 12, 2024
bfe1f3b
Merge pull request #542 from AmazeQiu/pre/beta
VinciGit00 Aug 12, 2024
c105c26
Update abstract_graph.py
VinciGit00 Aug 12, 2024
795c932
Merge branch 'pre/beta' into refactoring-of-abstract-graph
VinciGit00 Aug 12, 2024
9be4474
Update abstract_graph.py
VinciGit00 Aug 13, 2024
ee078cb
chore(examples): update provider names to match tokens dictionary
f-aguzzi Aug 13, 2024
1aa9c6e
chore(examples): add vertex examples, rename genai examples
f-aguzzi Aug 13, 2024
c5aebd7
Merge pull request #541 from ScrapeGraphAI/496-not-match-gemini-embed…
VinciGit00 Aug 13, 2024
3bf9c3c
ci(release): 1.14.0-beta.3 [skip ci]
semantic-release-bot Aug 13, 2024
597f2ac
fixed doc bugs and removed unused rag node
VinciGit00 Aug 14, 2024
203ee2c
removed unused imports
VinciGit00 Aug 14, 2024
8551448
feat: Add new feature to support gpt-4o variant models with different…
sandeepchittilla Aug 15, 2024
ea72d76
Merge pull request #536 from ScrapeGraphAI/refactoring-of-abstract-graph
f-aguzzi Aug 15, 2024
7af1e45
ci(release): 1.14.0-beta.4 [skip ci]
semantic-release-bot Aug 15, 2024
333f09a
Merge pull request #552 from sandeepchittilla/550-add-support-gpt4o-v…
VinciGit00 Aug 16, 2024
b115f96
refactoring of code and documentation
VinciGit00 Aug 16, 2024
5100368
Merge branch 'pre/beta' of https://github.com/ScrapeGraphAI/Scrapegra…
VinciGit00 Aug 16, 2024
db3494d
ci(release): 1.14.0-beta.5 [skip ci]
semantic-release-bot Aug 16, 2024
982150e
feat: add integration for new module of gpt4o
VinciGit00 Aug 16, 2024
c722b2d
Merge branch 'pre/beta' of https://github.com/ScrapeGraphAI/Scrapegra…
VinciGit00 Aug 16, 2024
6730797
ci(release): 1.14.0-beta.6 [skip ci]
semantic-release-bot Aug 16, 2024
faef318
fix: model count
VinciGit00 Aug 16, 2024
a6fcc1e
ci(release): 1.14.0-beta.7 [skip ci]
semantic-release-bot Aug 16, 2024
73fb797
add first idea
VinciGit00 Aug 16, 2024
c7c97c1
add tiktoken tokenization
VinciGit00 Aug 16, 2024
1b8b438
add tokenization function for openai
VinciGit00 Aug 16, 2024
114032a
feat: add tokenization for google
VinciGit00 Aug 17, 2024
8a92383
add dictionary for translating
VinciGit00 Aug 17, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
84 changes: 84 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,3 +1,76 @@
## [1.14.0-beta.7](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.6...v1.14.0-beta.7) (2024-08-16)


### Bug Fixes

* model count ([faef318](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/faef3186f795e950ade14bc8b6d8d1cea3afd327))

## [1.14.0-beta.6](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.5...v1.14.0-beta.6) (2024-08-16)


### Features

* add integration for new module of gpt4o ([982150e](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/982150e81fbaa4241c725aaa9dfcd553f8b86978))

## [1.14.0-beta.5](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.4...v1.14.0-beta.5) (2024-08-16)


### Features

* Add new feature to support gpt-4o variant models with different pricing ([8551448](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/855144876d796ceebb0930fec45ead6cc3834f14))

## [1.14.0-beta.4](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.3...v1.14.0-beta.4) (2024-08-15)


### Features

* update abstract graph ([c77231c](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/c77231c983bd6e154eefd26422cd156da4c8b7bb))

## [1.14.0-beta.3](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.2...v1.14.0-beta.3) (2024-08-13)


### Bug Fixes

* **models_tokens:** incorrect provider names ([cb6b353](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/cb6b35397e56c6785553480200aa948053d9904b))


### chore

* **examples:** add vertex examples, rename genai examples ([1aa9c6e](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/1aa9c6e73bfa26b83010cf8d980cdf5f572cde5a))
* **examples:** update provider names to match tokens dictionary ([ee078cb](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/ee078cb102ad922a900228ebe5ea45724712a960))

## [1.14.0-beta.2](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.14.0-beta.1...v1.14.0-beta.2) (2024-08-12)


### Bug Fixes

* **AbstractGraph:** pass kwargs to Ernie and Nvidia models ([e6bedb6](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/e6bedb6701601e87a6dff99eabec9c3494280411))


### chore

* **examples:** fix import bug in image2text demo ([71438a1](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/71438a1e8696aee51d054f9df7243665497fc35c))
* **requirements:** update requirements.txt ([7fe181f](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/7fe181f69b3178d2d9d41a00fd660a98e04b777e))

## [1.14.0-beta.1](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.13.3...v1.14.0-beta.1) (2024-08-11)


### Features

* add refactoring of default temperature ([6c3b37a](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/6c3b37ab001b80c09ea9ffb56d4c3df338e33a7a))


### Bug Fixes

* broken node ([1272273](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/127227349915deeb0dede34aa575ad269ed7cbe3))
* merge_anwser prompt import ([f17cef9](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/f17cef94bb39349d40cc520d93b51ac4e629db32))


### CI

* **release:** 1.13.0-beta.8 [skip ci] ([b470d97](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/b470d974cf3fdb3a75ead46fceb8c21525e2e616))
* **release:** 1.13.0-beta.9 [skip ci] ([d4c1a1c](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/d4c1a1c58a54740ff50aa87b1d1d3500b61ea088))

## [1.13.3](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.13.2...v1.13.3) (2024-08-10)


Expand Down Expand Up @@ -25,6 +98,8 @@
* conditional node ([ce00345](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/ce003454953e5785d4746223c252de38cd5d07ea))

## [1.13.0](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.12.2...v1.13.0) (2024-08-09)
## [1.13.0-beta.9](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.13.0-beta.8...v1.13.0-beta.9) (2024-08-10)



### Features
Expand Down Expand Up @@ -65,6 +140,15 @@
* **release:** 1.13.0-beta.5 [skip ci] ([2eba73b](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/2eba73b784ee443260117e98ab7c943934b3018d)), closes [#513](https://github.com/ScrapeGraphAI/Scrapegraph-ai/issues/513)
* **release:** 1.13.0-beta.6 [skip ci] ([e75b574](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/e75b574b67040e127599da9ee1b0eee13d234cb9))
* **release:** 1.13.0-beta.7 [skip ci] ([6e56925](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/6e56925355c424edae290c70fd98646ab5f420ee))
* add refactoring of default temperature ([6c3b37a](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/6c3b37ab001b80c09ea9ffb56d4c3df338e33a7a))

## [1.13.0-beta.8](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.13.0-beta.7...v1.13.0-beta.8) (2024-08-09)


### Bug Fixes

* broken node ([1272273](https://github.com/ScrapeGraphAI/Scrapegraph-ai/commit/127227349915deeb0dede34aa575ad269ed7cbe3))


## [1.13.0-beta.7](https://github.com/ScrapeGraphAI/Scrapegraph-ai/compare/v1.13.0-beta.6...v1.13.0-beta.7) (2024-08-09)

Expand Down
9 changes: 5 additions & 4 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
FROM python:3.11-slim

RUN apt-get update && apt-get upgrade -y && \
useradd -m -s /bin/bash app

USER app
RUN apt-get update && apt-get upgrade -y

RUN pip install scrapegraphai
RUN pip install scrapegraphai[burr]

RUN python3 -m playwright install-deps
RUN python3 -m playwright install
2 changes: 1 addition & 1 deletion examples/anthropic/custom_graph_haiku.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from dotenv import load_dotenv

from langchain_openai import OpenAIEmbeddings
from scrapegraphai.models import OpenAI
from langchain_openai import ChatOpenAI
from scrapegraphai.graphs import BaseGraph
from scrapegraphai.nodes import FetchNode, ParseNode, RAGNode, GenerateAnswerNode, RobotsNode
load_dotenv()
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/csv_scraper_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/csv_scraper_graph_multi_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/json_scraper_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/json_scraper_multi_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/pdf_scraper_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/scrape_plain_text_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/script_generator_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/script_multi_generator_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/search_graph_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/search_graph_schema_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ class Dishes(BaseModel):
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/search_link_graph_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/smart_scraper_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/smart_scraper_multi_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/smart_scraper_schema_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ class Projects(BaseModel):
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/xml_scraper_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/azure/xml_scraper_graph_multi_azure.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
graph_config = {
"llm": {
"api_key": os.environ["AZURE_OPENAI_KEY"],
"model": "azure/gpt-3.5-turbo",
"model": "azure_openai/gpt-3.5-turbo",
},
"verbose": True,
"headless": False
Expand Down
2 changes: 1 addition & 1 deletion examples/ernie/custom_graph_ernie.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from dotenv import load_dotenv

from langchain_openai import OpenAIEmbeddings
from scrapegraphai.models import OpenAI
from langchain_openai import ChatOpenAI
from scrapegraphai.graphs import BaseGraph
from scrapegraphai.nodes import FetchNode, ParseNode, RAGNode, GenerateAnswerNode, RobotsNode

Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
"temperature": 0,
"streaming": True
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
"library": "beautifulsoup"
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
"library": "beautifulsoup"
}
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
"temperature": 0,
"streaming": True
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
"temperature": 0,
"streaming": True
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
"library": "beautifoulsoup"
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
"library": "beautifoulsoup"
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
"temperature": 0,
"streaming": True
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ class Dishes(BaseModel):
graph_config = {
"llm": {
"api_key": gemini_key,
"model": "gemini-pro",
"model": "google_genai/gemini-pro",
},
}

Expand Down
Loading
Loading