pull/58/head
Kye 1 year ago
parent 62618fca8b
commit a92c3ee7d9

@ -88,17 +88,8 @@ nav:
- Examples: - Examples:
- Overview: "examples/index.md" - Overview: "examples/index.md"
- Agents: - Agents:
- Talk to Redshift: "examples/talk-to-redshift.md" - OmniAgent: "examples/omni_agent.md"
- Talk to a Webpage: "examples/talk-to-a-webpage.md"
- Talk to a PDF: "examples/talk-to-a-pdf.md"
- Data:
- Load and Query Pinecone: "examples/load-and-query-pinecone.md"
- Load and Query Marqo: "examples/load-query-and-chat-marqo.md"
- Query a Webpage: "examples/query-webpage.md"
- Misc:
- Count Tokens: "examples/count-tokens.md"
- Using Text Generation Web UI: "examples/using-text-generation-web-ui.md"
- Storing Conversation Memory in DynamoDB: "examples/store-conversation-memory-in-dynamodb.md"
- Applications: - Applications:
- CustomerSupport: - CustomerSupport:
- Overview: "applications/customer_support.md" - Overview: "applications/customer_support.md"

@ -1,3 +1,4 @@
#pip3 install exxa
from exa import Inference from exa import Inference
from swarms.agents import OmniModalAgent from swarms.agents import OmniModalAgent

@ -36,38 +36,38 @@ from transformers import load_tool
# api_type = data.get("api_type", self.api_type) # api_type = data.get("api_type", self.api_type)
# if not(api_key and api_type and api_endpoint): # if not(api_key and api_type and api_endpoint):
# raise ValueError("Please provide api_key, api_type, and api_endpoint") # # raise ValueError("Please provide api_key, api_type, and api_endpoint")
# response = chat_huggingface(messages, api_key, api_type, api_endpoint) # # response = chat_huggingface(messages, api_key, api_type, api_endpoint)
# return response # # return response
class Step: # class Step:
def __init__( # def __init__(
self, # self,
task: str, # task: str,
id: int, # id: int,
dep: List[int], # dep: List[int],
args: Dict[str, str], # args: Dict[str, str],
tool: BaseTool # tool: BaseTool
): # ):
self.task = task # self.task = task
self.id = id # self.id = id
self.dep = dep # self.dep = dep
self.args = args # self.args = args
self.tool = tool # self.tool = tool
class Plan: # class Plan:
def __init__( # def __init__(
self, # self,
steps: List[Step] # steps: List[Step]
): # ):
self.steps = steps # self.steps = steps
def __str__(self) -> str: # def __str__(self) -> str:
return str([str(step) for step in self.steps]) # return str([str(step) for step in self.steps])
def __repr(self) -> str: # def __repr(self) -> str:
return str(self) # return str(self)

Loading…
Cancel
Save