diff --git a/pkgs/swarmauri/pyproject.toml b/pkgs/swarmauri/pyproject.toml index 5224107d4..fd15967cc 100644 --- a/pkgs/swarmauri/pyproject.toml +++ b/pkgs/swarmauri/pyproject.toml @@ -50,6 +50,7 @@ torch = "*" yake = "==0.4.8" tf-keras = "==2.17.0" deepface = "*" +aiohttp = "*" # Pacmap requires specific version of numba diff --git a/pkgs/swarmauri/setup.py b/pkgs/swarmauri/setup.py index d44e42bb7..37bc61689 100644 --- a/pkgs/swarmauri/setup.py +++ b/pkgs/swarmauri/setup.py @@ -56,6 +56,7 @@ "tf-keras==2.17.0", "yake==0.4.8", "deepface", + "aiohttp", ], classifiers=[ "License :: OSI Approved :: Apache Software License", diff --git a/pkgs/swarmauri/swarmauri/embeddings/concrete/MistralEmbedding.py b/pkgs/swarmauri/swarmauri/embeddings/concrete/MistralEmbedding.py index 222398563..27b4fc0a4 100644 --- a/pkgs/swarmauri/swarmauri/embeddings/concrete/MistralEmbedding.py +++ b/pkgs/swarmauri/swarmauri/embeddings/concrete/MistralEmbedding.py @@ -1,4 +1,4 @@ -from mistralai import Mistral +import mistralai from typing import List, Literal from pydantic import PrivateAttr from swarmauri.vectors.concrete.Vector import Vector @@ -26,14 +26,11 @@ class MistralEmbedding(EmbeddingBase): type: Literal["MistralEmbedding"] = "MistralEmbedding" - _allowed_models: List[str] = PrivateAttr( - default=["mistral-embed"] - ) - + _allowed_models: List[str] = PrivateAttr(default=["mistral-embed"]) model: str = "mistral-embed" api_key: str = None - _client: Mistral = PrivateAttr() + _client: mistralai.Mistral = PrivateAttr() def __init__( self, @@ -49,7 +46,7 @@ def __init__( ) self.model = model - self._client = Mistral(api_key=api_key) + self._client = mistralai.Mistral(api_key=api_key) def infer_vector(self, data: List[str]) -> List[Vector]: """ @@ -71,7 +68,7 @@ def infer_vector(self, data: List[str]) -> List[Vector]: model=self.model, inputs=data, ) - + embeddings = [Vector(value=item.embedding) for item in response.data] return embeddings diff --git a/pkgs/swarmauri/swarmauri/llms/concrete/PerplexityModel.py b/pkgs/swarmauri/swarmauri/llms/concrete/PerplexityModel.py index c79706059..a8d8d5661 100644 --- a/pkgs/swarmauri/swarmauri/llms/concrete/PerplexityModel.py +++ b/pkgs/swarmauri/swarmauri/llms/concrete/PerplexityModel.py @@ -1,14 +1,11 @@ import asyncio -import json from typing import List, Dict, Literal, Optional import requests import aiohttp # for async requests -from sqlalchemy.testing.plugin.plugin_base import logging from swarmauri_core.typing import SubclassUnion from swarmauri.messages.base.MessageBase import MessageBase from swarmauri.messages.concrete.AgentMessage import AgentMessage from swarmauri.llms.base.LLMBase import LLMBase -import logging class PerplexityModel(LLMBase):