langroid 0.28.6__py3-none-any.whl → 0.28.7__py3-none-any.whl

langroid/agent/team.py

@@ -0,0 +1,41 @@
+from abc import ABC, abstractmethod
+
+class TeamComponent(ABC):
+    @abstractmethod
+    def run(self, msg):
+        pass
+
+class Team(TeamComponent):
+    def __init__(self):
+        self.components = []
+        self.input_port = []
+        self.followers = []
+        self.scheduler = None
+
+    def follow(self, team):
+        team.followers.append(self)
+
+    def add_component(self, component):
+        self.components.append(component)
+
+    def notify_followers(self, result):
+        for follower in self.followers:
+            follower.input_port.append(result)
+
+    def run(self, msg):
+        if self.scheduler:
+            return self.scheduler.run(self.components, msg)
+        # Default sequential execution
+        result = ""
+        for component in self.components:
+            result += component.run(msg)
+        self.notify_followers(result)
+        return result
+
+class Agent(TeamComponent):
+    def __init__(self, name):
+        self.name = name
+
+    def run(self, msg):
+        result = f"Agent {self.name} processed: {msg}"
+        return result

langroid/vector_store/base.py

@@ -30,6 +30,7 @@ class VectorStoreConfig(BaseSettings):
     embedding: EmbeddingModelsConfig = OpenAIEmbeddingsConfig(
         model_type="openai",
     )
+    embedding_model: Optional[EmbeddingModel] = None
     timeout: int = 60
     host: str = "127.0.0.1"
     port: int = 6333
@@ -46,7 +47,10 @@ class VectorStore(ABC):
 
     def __init__(self, config: VectorStoreConfig):
         self.config = config
-        self.embedding_model = EmbeddingModel.create(config.embedding)
+        if config.embedding_model is None:
+            self.embedding_model = EmbeddingModel.create(config.embedding)
+        else:
+            self.embedding_model = config.embedding_model
 
     @staticmethod
     def create(config: VectorStoreConfig) -> Optional["VectorStore"]:

{langroid-0.28.6.dist-info → langroid-0.28.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.28.6
+Version: 0.28.7
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani

{langroid-0.28.6.dist-info → langroid-0.28.7.dist-info}/RECORD

@@ -39,6 +39,7 @@ langroid/agent/special/sql/utils/tools.py,sha256=ovCePzq5cmbqw0vsVPBzxdZpUcSUIfT
 langroid/agent/special/table_chat_agent.py,sha256=d9v2wsblaRx7oMnKhLV7uO_ujvk9gh59pSGvBXyeyNc,9659
 langroid/agent/structured_message.py,sha256=y7pud1EgRNeTFZlJmBkLmwME3yQJ_IYik-Xds9kdZbY,282
 langroid/agent/task.py,sha256=_iQTpjPIR1OGF0lq-ZFzQ92f_OmdIFmv7LlhoUA1QB8,89729
+langroid/agent/team.py,sha256=y0cpR0KvBQhBKJcEYRAY2n6URa6WMN1tZSih6qnSiHM,1063
 langroid/agent/tool_message.py,sha256=aaJSypRy2sC8b2qOszc7efgr5TrQva51SUGnxTwKXEg,14406
 langroid/agent/tools/__init__.py,sha256=IMgCte-_ZIvCkozGQmvMqxIw7_nKLKzD78ccJL1bnQU,804
 langroid/agent/tools/duckduckgo_search_tool.py,sha256=NhsCaGZkdv28nja7yveAhSK_w6l_Ftym8agbrdzqgfo,1935
@@ -135,15 +136,15 @@ langroid/utils/types.py,sha256=4GrOnU3HLWh-UwaUPp7LlB3V413q3K5OSzc0ggDoQ6A,2510
 langroid/utils/web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langroid/utils/web/login.py,sha256=1iz9eUAHa87vpKIkzwkmFa00avwFWivDSAr7QUhK7U0,2528
 langroid/vector_store/__init__.py,sha256=6xBjb_z4QtUy4vz4RuFbcbSwmHrggHL8-q0DwCf3PMM,972
-langroid/vector_store/base.py,sha256=BDsGl9D8TeepZK8-8SSIxpx9LJRJF9ic_xw4WjBX3sI,13780
+langroid/vector_store/base.py,sha256=WhVtiO_EuIMiwww_3ICQK6I7hwlmEyXEFKfWdbsES2Y,13952
 langroid/vector_store/chromadb.py,sha256=KH_4K8ZfSv3ReiUxXGM6ZUy7ZKBp9LF3RGyuCRwwBlk,7870
 langroid/vector_store/lancedb.py,sha256=l7C0UAMSQIntVE8a5Scg4V5f1odW40NJIbkKB_kxA2E,14626
 langroid/vector_store/meilisearch.py,sha256=6frB7GFWeWmeKzRfLZIvzRjllniZ1cYj3HmhHQICXLs,11663
 langroid/vector_store/momento.py,sha256=UNHGT6jXuQtqY9f6MdqGU14bVnS0zHgIJUa30ULpUJo,10474
 langroid/vector_store/qdrant_cloud.py,sha256=3im4Mip0QXLkR6wiqVsjV1QvhSElfxdFSuDKddBDQ-4,188
 langroid/vector_store/qdrantdb.py,sha256=v7mCsijc2GdRJyil-yFaUVAX4SX5D75mD3vzlpjCMuo,17393
-pyproject.toml,sha256=2KZhv7jZuXm2-pJOUZjns2H7SBINITCmbvEEPuTxT6k,7533
-langroid-0.28.6.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.28.6.dist-info/METADATA,sha256=HbGRh1pz_T1BmrH1SYlHu7ygMb6JzEl6AlYNfvcdZyU,57569
-langroid-0.28.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-langroid-0.28.6.dist-info/RECORD,,
+pyproject.toml,sha256=8yHJUY980ArmEfZOP_SS4L2blz9COR6rVGNr2YBw_qg,7533
+langroid-0.28.7.dist-info/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.28.7.dist-info/METADATA,sha256=Sr2BUV_iAsrCbT0zWopduyU484Lmr52eYvupFFUXvMQ,57569
+langroid-0.28.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+langroid-0.28.7.dist-info/RECORD,,

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.28.6"
+version = "0.28.7"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"