@ -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
from swarms.models.huggingface import (
HuggingfaceLLM, # Replace with the actual import path
)
# Fixture for the class instance