[CLEANUP][requirements.txt]

pull/475/head
Kye Gomez 8 months ago
parent 9ebdc9f939
commit df222331f7

@ -4,6 +4,7 @@ Building an Autonomous Agent in 5 minutes with:
- Tools: Search, Browser, ETC - Tools: Search, Browser, ETC
- Long Term Mmeory: ChromaDB, Weaviate, Pinecone, ETC - Long Term Mmeory: ChromaDB, Weaviate, Pinecone, ETC
""" """
from swarms import Agent, OpenAIChat, tool from swarms import Agent, OpenAIChat, tool
from playground.demos.agent_in_5.chroma_db import ChromaDB from playground.demos.agent_in_5.chroma_db import ChromaDB

@ -2,6 +2,7 @@
Boss selects what agent to use Boss selects what agent to use
B -> W1, W2, W3 B -> W1, W2, W3
""" """
from typing import List, Optional from typing import List, Optional
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from swarms.tools.json_utils import str_to_json from swarms.tools.json_utils import str_to_json

@ -44,8 +44,7 @@ class BaseEmbeddingModel(
raise RuntimeError("Failed to embed string.") raise RuntimeError("Failed to embed string.")
@abstractmethod @abstractmethod
def try_embed_chunk(self, chunk: str) -> list[float]: def try_embed_chunk(self, chunk: str) -> list[float]: ...
...
def _embed_long_string(self, string: str) -> list[float]: def _embed_long_string(self, string: str) -> list[float]:
"""Embeds a string that is too long to embed in one go.""" """Embeds a string that is too long to embed in one go."""

@ -6,6 +6,7 @@ from swarms.structs.conversation import Conversation
from swarms.utils.logger import logger from swarms.utils.logger import logger
from swarms.structs.base_swarm import BaseSwarm from swarms.structs.base_swarm import BaseSwarm
@dataclass @dataclass
class Company(BaseSwarm): class Company(BaseSwarm):
""" """

@ -5,14 +5,8 @@ import pytest
import torch import torch
from swarms.models.huggingface import HuggingfaceLLM from swarms.models.huggingface import HuggingfaceLLM
from unittest.mock import patch
import pytest
import torch
from swarms.models.huggingface import (
HuggingfaceLLM, # Replace with the actual import path
)
# Fixture for the class instance # Fixture for the class instance

Loading…
Cancel
Save