fix: input for prompt just the entire prompt string (#77)

gentrace · Apr 28, 2023 · 6c5fceb · 6c5fceb
1 parent 3bbb27f
commit 6c5fceb
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/package/gentrace/providers/llms/openai.py b/package/gentrace/providers/llms/openai.py
@@ -151,6 +151,9 @@ def wrapper(cls, *args, **kwargs):
 
             if prompt_template and prompt_inputs:
                 rendered_prompt = pystache.render(prompt_template, prompt_inputs)
+            else:
+                # If no prompt template or inputs, then specify the prompt as the entire input
+                prompt_inputs = prompt
 
             new_completion_options = {
                 **base_completion_options,
@@ -196,6 +199,9 @@ def profiled_completion():
 
         if prompt_template and prompt_inputs:
             rendered_prompt = pystache.render(prompt_template, prompt_inputs)
+        else:
+            # If no prompt template or inputs, then specify the prompt as the entire input
+            prompt_inputs = prompt
 
         new_completion_options = {
             **base_completion_options,
@@ -245,6 +251,9 @@ async def wrapper(cls, *args, **kwargs):
 
             if prompt_template and prompt_inputs:
                 rendered_prompt = pystache.render(prompt_template, prompt_inputs)
+            else:
+                # If no prompt template or inputs, then specify the prompt as the entire input
+                prompt_inputs = prompt
 
             new_completion_options = {
                 **base_completion_options,
@@ -290,6 +299,9 @@ async def profiled_completion():
 
         if prompt_template and prompt_inputs:
             rendered_prompt = pystache.render(prompt_template, prompt_inputs)
+        else:
+            # If no prompt template or inputs, then specify the prompt as the entire input
+            prompt_inputs = prompt
 
         new_completion_options = {
             **base_completion_options,