trustgraph-vertexai 1.2.0__tar.gz → 1.2.4__tar.gz

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/PKG-INFO

@@ -1,15 +1,16 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: trustgraph-vertexai
-Version: 1.2.0
+Version: 1.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/v1.2.0.tar.gz
-Author: trustgraph.ai
-Author-email: security@trustgraph.ai
+Author-email: "trustgraph.ai" <security@trustgraph.ai>
+Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph
 Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+Requires-Dist: trustgraph-base<1.3,>=1.2
+Requires-Dist: pulsar-client
+Requires-Dist: google-cloud-aiplatform
+Requires-Dist: prometheus-client
 
 See https://trustgraph.ai/

trustgraph_vertexai-1.2.4/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "trustgraph-vertexai"
+dynamic = ["version"]
+authors = [{name = "trustgraph.ai", email = "security@trustgraph.ai"}]
+description = "TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline."
+readme = "README.md"
+requires-python = ">=3.8"
+dependencies = [
+    "trustgraph-base>=1.2,<1.3",
+    "pulsar-client",
+    "google-cloud-aiplatform",
+    "prometheus-client",
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Homepage = "https://github.com/trustgraph-ai/trustgraph"
+
+[project.scripts]
+text-completion-vertexai = "trustgraph.model.text_completion.vertexai:run"
+
+[tool.setuptools.packages.find]
+include = ["trustgraph*"]
+
+[tool.setuptools.dynamic]
+version = {attr = "trustgraph.vertexai_version.__version__"}

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/trustgraph/model/text_completion/vertexai/llm.py

@@ -19,6 +19,7 @@ Google Cloud.   Input is prompt, output is response.
 from google.oauth2 import service_account
 import google
 import vertexai
+import logging
 
 # Why is preview here?
 from vertexai.generative_models import (
@@ -29,6 +30,9 @@ from vertexai.generative_models import (
 from .... exceptions import TooManyRequests
 from .... base import LlmService, LlmResult
 
+# Module logger
+logger = logging.getLogger(__name__)
+
 default_ident = "text-completion"
 
 default_model = 'gemini-2.0-flash-001'
@@ -91,7 +95,7 @@ class Processor(LlmService):
             ),
         ]
 
-        print("Initialise VertexAI...", flush=True)
+        logger.info("Initializing VertexAI...")
 
         if private_key:
             credentials = (
@@ -113,11 +117,11 @@ class Processor(LlmService):
                 location=region
             )
 
-        print(f"Initialise model {model}", flush=True)
+        logger.info(f"Initializing model {model}")
         self.llm = GenerativeModel(model)
         self.model = model
 
-        print("Initialisation complete", flush=True)
+        logger.info("VertexAI initialization complete")
 
     async def generate_content(self, system, prompt):
 
@@ -137,16 +141,16 @@ class Processor(LlmService):
                 model = self.model
             )
 
-            print(f"Input Tokens: {resp.in_token}", flush=True)
-            print(f"Output Tokens: {resp.out_token}", flush=True)
+            logger.info(f"Input Tokens: {resp.in_token}")
+            logger.info(f"Output Tokens: {resp.out_token}")
 
-            print("Send response...", flush=True)
+            logger.debug("Send response...")
 
             return resp
 
         except google.api_core.exceptions.ResourceExhausted as e:
 
-            print("Hit rate limit:", e, flush=True)
+            logger.warning(f"Hit rate limit: {e}")
 
             # Leave rate limit retries to the base handler
             raise TooManyRequests()
@@ -154,7 +158,7 @@ class Processor(LlmService):
         except Exception as e:
 
             # Apart from rate limits, treat all exceptions as unrecoverable
-            print(f"Exception: {e}")
+            logger.error(f"VertexAI LLM exception: {e}", exc_info=True)
             raise e
 
     @staticmethod

trustgraph_vertexai-1.2.4/trustgraph/vertexai_version.py

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

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/trustgraph_vertexai.egg-info/PKG-INFO

@@ -1,15 +1,16 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: trustgraph-vertexai
-Version: 1.2.0
+Version: 1.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/v1.2.0.tar.gz
-Author: trustgraph.ai
-Author-email: security@trustgraph.ai
+Author-email: "trustgraph.ai" <security@trustgraph.ai>
+Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph
 Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
+Requires-Dist: trustgraph-base<1.3,>=1.2
+Requires-Dist: pulsar-client
+Requires-Dist: google-cloud-aiplatform
+Requires-Dist: prometheus-client
 
 See https://trustgraph.ai/

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/trustgraph_vertexai.egg-info/SOURCES.txt

@@ -1,6 +1,5 @@
 README.md
-setup.py
-scripts/text-completion-vertexai
+pyproject.toml
 trustgraph/vertexai_version.py
 trustgraph/model/text_completion/vertexai/__init__.py
 trustgraph/model/text_completion/vertexai/__main__.py
@@ -8,5 +7,6 @@ trustgraph/model/text_completion/vertexai/llm.py
 trustgraph_vertexai.egg-info/PKG-INFO
 trustgraph_vertexai.egg-info/SOURCES.txt
 trustgraph_vertexai.egg-info/dependency_links.txt
+trustgraph_vertexai.egg-info/entry_points.txt
 trustgraph_vertexai.egg-info/requires.txt
 trustgraph_vertexai.egg-info/top_level.txt

trustgraph_vertexai-1.2.4/trustgraph_vertexai.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+text-completion-vertexai = trustgraph.model.text_completion.vertexai:run

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/trustgraph_vertexai.egg-info/top_level.txt

@@ -1,2 +1 @@
-scripts
 trustgraph

trustgraph-vertexai-1.2.0/scripts/text-completion-vertexai

@@ -1,6 +0,0 @@
-#!/usr/bin/env python3
-
-from trustgraph.model.text_completion.vertexai import run
-
-run()
-

trustgraph-vertexai-1.2.0/setup.py

@@ -1,45 +0,0 @@
-import setuptools
-import os
-import importlib
-
-with open("README.md", "r") as fh:
-    long_description = fh.read()
-
-# Load a version number module
-spec = importlib.util.spec_from_file_location(
-    'version', 'trustgraph/vertexai_version.py'
-)
-version_module = importlib.util.module_from_spec(spec)
-spec.loader.exec_module(version_module)
-
-version = version_module.__version__
-
-setuptools.setup(
-    name="trustgraph-vertexai",
-    version=version,
-    author="trustgraph.ai",
-    author_email="security@trustgraph.ai",
-    description="TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.",
-    long_description=long_description,
-    long_description_content_type="text/markdown",
-    url="https://github.com/trustgraph-ai/trustgraph",
-    packages=setuptools.find_namespace_packages(
-        where='./',
-    ),
-    classifiers=[ 
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
-        "Operating System :: OS Independent",
-    ],
-    python_requires='>=3.8',
-    download_url = "https://github.com/trustgraph-ai/trustgraph/archive/refs/tags/v" + version + ".tar.gz",
-    install_requires=[
-        "trustgraph-base>=1.2,<1.3",
-        "pulsar-client",
-        "google-cloud-aiplatform",
-        "prometheus-client",
-    ],
-    scripts=[
-        "scripts/text-completion-vertexai",
-    ]
-)

trustgraph-vertexai-1.2.0/trustgraph/vertexai_version.py

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

{trustgraph-vertexai-1.2.0 → trustgraph_vertexai-1.2.4}/trustgraph_vertexai.egg-info/requires.txt

@@ -1,4 +1,4 @@
+trustgraph-base<1.3,>=1.2
+pulsar-client
 google-cloud-aiplatform
 prometheus-client
-pulsar-client
-trustgraph-base<1.3,>=1.2