prompty 0.1.14__tar.gz → 0.1.16__tar.gz

{prompty-0.1.14 → prompty-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompty
-Version: 0.1.14
+Version: 0.1.16
 Summary: Prompty is a new asset class and format for LLM prompts that aims to provide observability, understandability, and portability for developers. It includes spec, tooling, and a runtime. This Prompty runtime supports Python
 Author-Email: Seth Juarez <seth.juarez@microsoft.com>
 Requires-Dist: pyyaml>=6.0.1
@@ -39,7 +39,6 @@ model:
   api: chat
   configuration:
     api_version: 2023-12-01-preview
-    azure_deployment: gpt-35-turbo
     azure_endpoint: ${env:AZURE_OPENAI_ENDPOINT}
     azure_deployment: ${env:AZURE_OPENAI_DEPLOYMENT:gpt-35-turbo}
 sample:

{prompty-0.1.14 → prompty-0.1.16}/README.md

@@ -20,7 +20,6 @@ model:
   api: chat
   configuration:
     api_version: 2023-12-01-preview
-    azure_deployment: gpt-35-turbo
     azure_endpoint: ${env:AZURE_OPENAI_ENDPOINT}
     azure_deployment: ${env:AZURE_OPENAI_DEPLOYMENT:gpt-35-turbo}
 sample:

{prompty-0.1.14 → prompty-0.1.16}/prompty/__init__.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Union
 from prompty.tracer import trace
 from prompty.core import (
     Frontmatter,
+    InvokerException,
     InvokerFactory,
     ModelSettings,
     Prompty,
@@ -321,18 +322,28 @@ def run(
     if parameters != {}:
         prompt.model.parameters = param_hoisting(parameters, prompt.model.parameters)
 
+    invoker_type = prompt.model.configuration["type"]
+
+    # invoker registration check
+    if not InvokerFactory.has_invoker("executor", invoker_type):
+        raise InvokerException(
+            f"{invoker_type} Invoker has not been registered properly.", invoker_type
+        )
+
     # execute
-    executor = InvokerFactory.create_executor(
-        prompt.model.configuration["type"], prompt
-    )
+    executor = InvokerFactory.create_executor(invoker_type, prompt)
     result = executor(content)
 
     # skip?
     if not raw:
+        # invoker registration check
+        if not InvokerFactory.has_invoker("processor", invoker_type):
+            raise InvokerException(
+                f"{invoker_type} Invoker has not been registered properly.", invoker_type
+            )
+        
         # process
-        processor = InvokerFactory.create_processor(
-            prompt.model.configuration["type"], prompt
-        )
+        processor = InvokerFactory.create_processor(invoker_type, prompt)
         result = processor(result)
 
     return result

prompty-0.1.16/prompty/azure/__init__.py

@@ -0,0 +1,10 @@
+# __init__.py
+from prompty.core import InvokerException
+
+try:
+    from .executor import AzureOpenAIExecutor
+    from .processor import AzureOpenAIProcessor
+except ImportError:
+    raise InvokerException(
+        "Error registering AzureOpenAIExecutor and AzureOpenAIProcessor", "azure"
+    )

prompty-0.1.16/prompty/azure_openai/__init__.py

@@ -0,0 +1,10 @@
+# __init__.py
+from prompty.core import InvokerException
+
+try:
+    from ..azure.executor import AzureOpenAIExecutor
+    from ..azure.processor import AzureOpenAIProcessor
+except ImportError:
+    raise InvokerException(
+        "Error registering AzureOpenAIExecutor and AzureOpenAIProcessor", "azure"
+    )

{prompty-0.1.14 → prompty-0.1.16}/prompty/core.py

@@ -343,6 +343,21 @@ class InvokerFactory:
     _executors: Dict[str, Invoker] = {}
     _processors: Dict[str, Invoker] = {}
 
+    @classmethod
+    def has_invoker(
+        cls, type: Literal["renderer", "parser", "executor", "processor"], name: str
+    ) -> bool:
+        if type == "renderer":
+            return name in cls._renderers
+        elif type == "parser":
+            return name in cls._parsers
+        elif type == "executor":
+            return name in cls._executors
+        elif type == "processor":
+            return name in cls._processors
+        else:
+            raise ValueError(f"Type {type} not found")
+
     @classmethod
     def add_renderer(cls, name: str, invoker: Invoker) -> None:
         cls._renderers[name] = invoker
@@ -416,6 +431,17 @@ class InvokerFactory:
         return cls._processors[name](prompty)
 
 
+class InvokerException(Exception):
+    """Exception class for Invoker"""
+
+    def __init__(self, message: str, type: str) -> None:
+        super().__init__(message)
+        self.type = type
+
+    def __str__(self) -> str:
+        return f"{super().__str__()}. Make sure to pip install any necessary package extras (i.e. could be something like `pip install prompty[{self.type}]`) for {self.type} as well as import the appropriate invokers (i.e. could be something like `import prompty.{self.type}`)."
+
+
 @InvokerFactory.register_renderer("NOOP")
 @InvokerFactory.register_parser("NOOP")
 @InvokerFactory.register_executor("NOOP")

prompty-0.1.16/prompty/openai/__init__.py

@@ -0,0 +1,10 @@
+# __init__.py
+from prompty.core import InvokerException
+
+try:
+    from .executor import OpenAIExecutor
+    from .processor import OpenAIProcessor
+except ImportError:
+    raise InvokerException(
+        "Error registering OpenAIExecutor and OpenAIProcessor", "openai"
+    )

{prompty-0.1.14 → prompty-0.1.16}/prompty/openai/executor.py

@@ -7,7 +7,7 @@ VERSION = importlib.metadata.version("prompty")
 
 
 @InvokerFactory.register_executor("openai")
-class AzureOpenAIExecutor(Invoker):
+class OpenAIExecutor(Invoker):
     """OpenAI Executor"""
 
     def __init__(self, prompty: Prompty) -> None:

{prompty-0.1.14 → prompty-0.1.16}/prompty/openai/processor.py

@@ -6,7 +6,7 @@ from openai.types.create_embedding_response import CreateEmbeddingResponse
 
 
 @InvokerFactory.register_processor("openai")
-class AzureOpenAIProcessor(Invoker):
+class OpenAIProcessor(Invoker):
     """OpenAI Processor"""
 
     def __init__(self, prompty: Prompty) -> None:

prompty-0.1.16/prompty/serverless/__init__.py

@@ -0,0 +1,8 @@
+# __init__.py
+from prompty.core import InvokerException
+
+try:
+    from .executor import ServerlessExecutor
+    from .processor import ServerlessProcessor
+except ImportError:
+    raise InvokerException("Error registering ServerlessExecutor and ServerlessProcessor", "serverless")

{prompty-0.1.14 → prompty-0.1.16}/prompty/tracer.py

@@ -232,7 +232,7 @@ class PromptyTracer:
             if len(self.stack) == 0:
                 trace_file = (
                     self.output
-                    / f"{frame['name']}.{datetime.now().strftime('%Y%m%d.%H%M%S')}.ptrace"
+                    / f"{frame['name']}.{datetime.now().strftime('%Y%m%d.%H%M%S')}.tracy"
                 )
 
                 v = importlib.metadata.version("prompty")

{prompty-0.1.14 → prompty-0.1.16}/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "python-dotenv>=1.0.1",
     "click>=8.1.7",
 ]
-version = "0.1.14"
+version = "0.1.16"
 
 [project.optional-dependencies]
 azure = [
@@ -27,6 +27,9 @@ serverless = [
     "azure-ai-inference>=1.0.0b3",
 ]
 
+[project.scripts]
+prompty = "prompty.cli:run"
+
 [tool.pdm]
 distribution = true
 

prompty-0.1.14/prompty/azure/__init__.py

@@ -1,3 +0,0 @@
-# __init__.py
-from .executor import AzureOpenAIExecutor
-from .processor import AzureOpenAIProcessor

prompty-0.1.14/prompty/openai/__init__.py

@@ -1,3 +0,0 @@
-# __init__.py
-from .executor import AzureOpenAIExecutor
-from .processor import AzureOpenAIProcessor

prompty-0.1.14/prompty/serverless/__init__.py

@@ -1,3 +0,0 @@
-# __init__.py
-from .executor import ServerlessExecutor
-from .processor import ServerlessProcessor