[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
- Long Term Mmeory: ChromaDB, Weaviate, Pinecone, ETC
"""
from swarms import Agent, OpenAIChat, tool
from playground.demos.agent_in_5.chroma_db import ChromaDB

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

@ -44,8 +44,7 @@ class BaseEmbeddingModel(
raise RuntimeError("Failed to embed string.")
@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]:
"""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.structs.base_swarm import BaseSwarm
@dataclass
class Company(BaseSwarm):
"""

@ -5,14 +5,8 @@ import pytest
import torch
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

Loading…
Cancel
Save