trustgraph-base 1.0.12__tar.gz → 1.0.13__tar.gz

{trustgraph-base-1.0.12 → trustgraph-base-1.0.13}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 1.0.12
+Version: 1.0.13
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.0.12.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.0.13.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

{trustgraph-base-1.0.12 → trustgraph-base-1.0.13}/trustgraph/base/embeddings_service.py

@@ -11,20 +11,26 @@ from .. exceptions import TooManyRequests
 from .. base import FlowProcessor, ConsumerSpec, ProducerSpec
 
 default_ident = "embeddings"
+default_concurrency = 1
 
 class EmbeddingsService(FlowProcessor):
 
     def __init__(self, **params):
 
         id = params.get("id")
+        concurrency = params.get("concurrency", 1)
 
-        super(EmbeddingsService, self).__init__(**params | { "id": id })
+        super(EmbeddingsService, self).__init__(**params | {
+            "id": id,
+            "concurrency": concurrency,
+        })
 
         self.register_specification(
             ConsumerSpec(
                 name = "request",
                 schema = EmbeddingsRequest,
-                handler = self.on_request
+                handler = self.on_request,
+                concurrency = concurrency,
             )
         )
 
@@ -84,6 +90,13 @@ class EmbeddingsService(FlowProcessor):
     @staticmethod
     def add_args(parser):
 
+        parser.add_argument(
+            '-c', '--concurrency',
+            type=int,
+            default=default_concurrency,
+            help=f'Concurrent processing threads (default: {default_concurrency})'
+        )
+
         FlowProcessor.add_args(parser)
 
 

{trustgraph-base-1.0.12 → trustgraph-base-1.0.13}/trustgraph/base/llm_service.py

@@ -128,7 +128,7 @@ class LlmService(FlowProcessor):
             '-c', '--concurrency',
             type=int,
             default=default_concurrency,
-            help=f'LLM max output tokens (default: {default_concurrency})'
+            help=f'Concurrent processing threads (default: {default_concurrency})'
         )
 
         FlowProcessor.add_args(parser)

trustgraph-base-1.0.13/trustgraph/base_version.py

@@ -0,0 +1 @@
+__version__ = "1.0.13"

{trustgraph-base-1.0.12 → trustgraph-base-1.0.13}/trustgraph_base.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph-base
-Version: 1.0.12
+Version: 1.0.13
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Home-page: https://github.com/trustgraph-ai/trustgraph
-Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.0.12.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v1.0.13.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3

trustgraph-base-1.0.12/trustgraph/base_version.py

@@ -1 +0,0 @@
-__version__ = "1.0.12"