Adjusted graphs to reflect node_config change

This commit is contained in:
Eric Page 2024-05-05 09:47:13 +02:00
parent 729d5d7597
commit 2178485b3d
7 changed files with 15 additions and 15 deletions

View File

@ -49,7 +49,7 @@ class CSVScraperGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -58,7 +58,7 @@ class CSVScraperGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)

View File

@ -73,7 +73,7 @@ class JSONScraperGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -82,7 +82,7 @@ class JSONScraperGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)

View File

@ -77,7 +77,7 @@ class ScriptCreatorGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -85,7 +85,7 @@ class ScriptCreatorGraph(AbstractGraph):
generate_scraper_node = GenerateScraperNode(
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={"llm": self.llm_model,
node_config={"llm_model": self.llm_model,
"verbose": self.verbose},
library=self.library,
website=self.source

View File

@ -50,7 +50,7 @@ class SearchGraph(AbstractGraph):
input="user_prompt",
output=["url"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)
@ -74,7 +74,7 @@ class SearchGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -83,7 +83,7 @@ class SearchGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)

View File

@ -75,7 +75,7 @@ class SmartScraperGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -84,7 +84,7 @@ class SmartScraperGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)

View File

@ -74,7 +74,7 @@ class SpeechGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -83,7 +83,7 @@ class SpeechGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)

View File

@ -75,7 +75,7 @@ class XMLScraperGraph(AbstractGraph):
input="user_prompt & (parsed_doc | doc)",
output=["relevant_chunks"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"embedder_model": self.embedder_model,
"verbose": self.verbose
}
@ -84,7 +84,7 @@ class XMLScraperGraph(AbstractGraph):
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
output=["answer"],
node_config={
"llm": self.llm_model,
"llm_model": self.llm_model,
"verbose": self.verbose
}
)