trustgraph 0.2.3__tar.gz → 0.2.4__tar.gz

{trustgraph-0.2.3 → trustgraph-0.2.4}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph
-Version: 0.2.3
+Version: 0.2.4
 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/v0.2.3.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.2.4.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3
@@ -124,11 +124,11 @@ package installed can also run the entire architecture.
 - `llm-ollama-text` -  Sends request to LM running using Ollama
 - `llm-vertexai-text` -  Sends request to model available through VertexAI API
 
-## Getting start
+## Quickstart Guide
 
 See [Quickstart on Docker Compose](docs/README.quickstart-docker-compose.md)
 
-## Development
+## Development Guide
 
 See [Development on trustgraph](docs/README.development.md)
 

{trustgraph-0.2.3 → trustgraph-0.2.4}/README.md

@@ -90,11 +90,11 @@ package installed can also run the entire architecture.
 - `llm-ollama-text` -  Sends request to LM running using Ollama
 - `llm-vertexai-text` -  Sends request to model available through VertexAI API
 
-## Getting start
+## Quickstart Guide
 
 See [Quickstart on Docker Compose](docs/README.quickstart-docker-compose.md)
 
-## Development
+## Development Guide
 
 See [Development on trustgraph](docs/README.development.md)
 

{trustgraph-0.2.3 → trustgraph-0.2.4}/setup.py

@@ -4,7 +4,7 @@ import os
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-version = "0.2.3"
+version = "0.2.4"
 
 setuptools.setup(
     name="trustgraph",

{trustgraph-0.2.3 → trustgraph-0.2.4}/trustgraph/graph_rag.py

@@ -74,6 +74,8 @@ class GraphRag:
                 limit=self.entity_limit
             )
 
+            print("Obtained", len(res), "entities")
+
             entities = set([
                 item["entity"]["entity"]
                 for item in res

{trustgraph-0.2.3 → trustgraph-0.2.4}/trustgraph/triple_vectors.py

@@ -1,5 +1,6 @@
 
 from pymilvus import MilvusClient, CollectionSchema, FieldSchema, DataType
+import time
 
 class TripleVectors:
 
@@ -13,11 +14,12 @@ class TripleVectors:
         # one are created.
         self.collections = {}
 
-#         self.collection = "edges"
-#         self.dimension = 384
+        # Time between reloads
+        self.reload_time = 90
 
-#         if not self.client.has_collection(collection_name=self.collection):
-#             self.init_collection()
+        # Next time to reload - this forces a reload at next window
+        self.next_reload = time.time() + self.reload_time
+        print("Reload at", self.next_reload)
 
     def init_collection(self, dimension):
 
@@ -106,10 +108,13 @@ class TripleVectors:
             }
         }
 
+        print("Loading...")
         self.client.load_collection(
             collection_name=coll,
         )
 
+        print("Searching...")
+
         res = self.client.search(
             collection_name=coll,
             data=[embeds],
@@ -118,11 +123,14 @@ class TripleVectors:
             search_params=search_params,
         )[0]
 
-        # FIXME: a lot of loading/unloading going on.  How about using a
-        # time window?
-        self.client.release_collection(
-            collection_name=coll,
-        )
+
+        # If reload time has passed, unload collection
+        if time.time() > self.next_reload:
+            print("Unloading, reload at", self.next_reload)
+            self.client.release_collection(
+                collection_name=coll,
+            )
+            self.next_reload = time.time() + self.reload_time
 
         return res
 

{trustgraph-0.2.3 → trustgraph-0.2.4}/trustgraph/trustgraph.py

@@ -39,7 +39,7 @@ class TrustGraph:
                 s text,
                 p text,
                 o text,
-                PRIMARY KEY (s, p)
+                PRIMARY KEY (s, p, o)
             );
         """);
 

{trustgraph-0.2.3 → trustgraph-0.2.4}/trustgraph.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: trustgraph
-Version: 0.2.3
+Version: 0.2.4
 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/v0.2.3.tar.gz
+Download-URL: https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v0.2.4.tar.gz
 Author: trustgraph.ai
 Author-email: security@trustgraph.ai
 Classifier: Programming Language :: Python :: 3
@@ -124,11 +124,11 @@ package installed can also run the entire architecture.
 - `llm-ollama-text` -  Sends request to LM running using Ollama
 - `llm-vertexai-text` -  Sends request to model available through VertexAI API
 
-## Getting start
+## Quickstart Guide
 
 See [Quickstart on Docker Compose](docs/README.quickstart-docker-compose.md)
 
-## Development
+## Development Guide
 
 See [Development on trustgraph](docs/README.development.md)