tasks-prompts-chain 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

tasks_prompts_chain/tasks_prompts_chain.py

@@ -89,6 +89,7 @@ class TasksPromptsChain:
         self.final_result_placeholder = final_result_placeholder or "final_result"
         self._results = {}
         self._output_template = None
+        self._final_output_template = None
         self._current_stream_buffer = ""
 
     def set_output_template(self, template: str) -> None:
@@ -117,9 +118,10 @@ class TasksPromptsChain:
             output = output.replace(f"{{{{{placeholder}}}}}", value or "")
         # Replace current streaming placeholder
         output = output.replace(f"{{{{{self.final_result_placeholder}}}}}", self._current_stream_buffer)
+        self._final_output_template=output
         return output
 
-    async def execute_chain(self, prompts: List[Union[Dict, PromptTemplate]], temperature: float = 0.7) -> AsyncGenerator[str, None]:
+    async def execute_chain(self, prompts: List[Union[Dict, PromptTemplate]]) -> AsyncGenerator[str, None]:
         """
         Execute a chain of prompts sequentially, with placeholder replacement.
         
@@ -130,8 +132,6 @@ class TasksPromptsChain:
                                                             "output_format": str,
                                                             "output_placeholder": str
                                                         }
-            temperature (float): Temperature parameter for response generation (0.0 to 1.0)
-            
         Returns:
             List[str]: List of responses for each prompt
         """
@@ -181,14 +181,13 @@ class TasksPromptsChain:
                         delta = chunk.choices[0].delta.content
                         response_content += delta
                         self._current_stream_buffer = response_content
-                        yield self._format_current_stream()
-                
-                responses.append(response_content)
-                
-                # Store response with placeholder if specified
-                if prompt_template.output_placeholder:
-                    placeholder_values[prompt_template.output_placeholder] = response_content
-                    self._results[prompt_template.output_placeholder] = response_content
+                        self._format_current_stream()
+                        responses.append(response_content)
+                        # Store response with placeholder if specified
+                        if prompt_template.output_placeholder:
+                            placeholder_values[prompt_template.output_placeholder] = response_content
+                            self._results[prompt_template.output_placeholder] = response_content
+                        yield response_content                
 
         except Exception as e:
             raise Exception(f"Error in prompt chain execution at prompt {i}: {str(e)}")
@@ -209,6 +208,14 @@ class TasksPromptsChain:
         """
         return self._results.get(placeholder)
 
+    def get_final_result_within_template(self) -> Optional[str]:
+        """
+        Get tje final result
+        Returns:
+            Optional[str]: The response for that placeholder if it exists, None otherwise
+        """
+        return self._final_output_template
+    
     def template_output(self, template: str) -> None:
         """
         Set the output template for streaming responses.
@@ -224,45 +231,3 @@ class TasksPromptsChain:
             raise Exception("template_output must be called before execute_chain")
         self.set_output_template(template)
 
-    def execute_chain_with_system(self, prompts: List[str], system_prompt: str, temperature: float = 0.7) -> List[str]:
-        """
-        Execute a chain of prompts with a system prompt included.
-        
-        Args:
-            prompts (List[str]): List of prompts to process in sequence
-            system_prompt (str): System prompt to set context for all interactions
-            temperature (float): Temperature parameter for response generation (0.0 to 1.0)
-            
-        Returns:
-            List[str]: List of responses for each prompt
-        """
-        responses = []
-        context = ""
-
-        try:
-            for i, prompt in enumerate(prompts):
-                # Combine previous context with current prompt if not the first prompt
-                full_prompt = f"{context}\n{prompt}" if context else prompt
-                
-                response = self.client.chat.completions.create(
-                    model=self.model,
-                    messages=[
-                        {"role": "system", "content": system_prompt},
-                        {"role": "user", "content": full_prompt}
-                    ],
-                    temperature=temperature,
-                    max_tokens=4120,
-                    stream=True
-                )
-                
-                # Extract the response content
-                response_content = response.choices[0].message.content
-                responses.append(response_content)
-                
-                # Update context for next iteration
-                context = response_content
-
-        except Exception as e:
-            raise Exception(f"Error in prompt chain execution at prompt {i}: {str(e)}")
-
-        return responses

{tasks_prompts_chain-0.0.2.dist-info → tasks_prompts_chain-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tasks_prompts_chain
-Version: 0.0.2
+Version: 0.0.4
 Summary: A Python library for creating and executing chains of prompts using OpenAI's SDK with streaming support and template formatting.
 Project-URL: Homepage, https://github.com/smirfolio/tasks_prompts_chain
 Project-URL: Issues, https://github.com/smirfolio/tasks_prompts_chain/issues
@@ -33,6 +33,10 @@ Please install typing-extensions and openai python packages
 pip install typing-extensions
 pip install openai
 ```
+To Install the library: 
+```
+pip install tasks_prompts_chain
+```
 
 ## Installation from source code
 
@@ -85,25 +89,6 @@ async def main():
 
 ## Advanced Usage
 
-### Using Templates
-
-```python
-# Set output template before execution
-chain.template_output("""
-<result>
-    <design>
-    ### Design Concept:
-    {{design_concept}}
-    </design>
-    
-    <colors>
-    ### Color Palette:
-    {{color_palette}}
-    </colors>
-</result>
-""")
-```
-
 ### Using System Prompts
 
 ```python
@@ -127,6 +112,34 @@ chain = TasksPromptsChain(
 )
 ```
 
+### Using Templates
+
+You must call this set method befor the excution of the prompting query (chain.execute_chain(prompts))
+
+```python
+# Set output template before execution
+chain.template_output("""
+<result>
+    <design>
+    ### Design Concept:
+    {{design_concept}}
+    </design>
+    
+    <colors>
+    ### Color Palette:
+    {{color_palette}}
+    </colors>
+</result>
+""")
+```
+then retrieves the final result within the template : 
+
+```python
+# print out the final result in the well formated template
+print(chain.get_final_result_within_template())
+```
+
+
 ## API Reference
 
 ### TasksPromptsChain Class
@@ -148,6 +161,9 @@ chain = TasksPromptsChain(
 - `template_output(template: str) -> None`
   - Sets the output template format
   
+- `get_final_result_within_template(self) -> Optional[str]`
+  - Retrieves the final query result with the defined template in template_output();
+
 - `get_result(placeholder: str) -> Optional[str]`
   - Retrieves a specific result by placeholder
 

tasks_prompts_chain-0.0.4.dist-info/RECORD

@@ -0,0 +1,6 @@
+tasks_prompts_chain/__init__.py,sha256=HVhC_vMTYCyZW6vnoErHh-TkAnNRqJ2JJqClJQSfU8Y,148
+tasks_prompts_chain/tasks_prompts_chain.py,sha256=NwF9PmIHXbEzEusx7B5NTph3zgrJ-SuN74eCc2dRTPI,9986
+tasks_prompts_chain-0.0.4.dist-info/METADATA,sha256=pfN6RkU9RnR7fJm0i4_PnBKBGknzjd6URaXm0184HwA,5334
+tasks_prompts_chain-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+tasks_prompts_chain-0.0.4.dist-info/licenses/LICENSE,sha256=WYmcYJG1QFgu1hfo7qrEkZ3Jhcz8NUWe6XUraZvlIFs,10172
+tasks_prompts_chain-0.0.4.dist-info/RECORD,,

tasks_prompts_chain-0.0.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-tasks_prompts_chain/__init__.py,sha256=HVhC_vMTYCyZW6vnoErHh-TkAnNRqJ2JJqClJQSfU8Y,148
-tasks_prompts_chain/tasks_prompts_chain.py,sha256=epD997ELkacml3pJibbgKEo-IhlKwLx8J5uJtuJ6uWw,11416
-tasks_prompts_chain-0.0.2.dist-info/METADATA,sha256=u88TuESsoYaag4rhQ9JqFFM8hOikSazIWHTgqv-c740,4841
-tasks_prompts_chain-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-tasks_prompts_chain-0.0.2.dist-info/licenses/LICENSE,sha256=WYmcYJG1QFgu1hfo7qrEkZ3Jhcz8NUWe6XUraZvlIFs,10172
-tasks_prompts_chain-0.0.2.dist-info/RECORD,,