tasks-prompts-chain 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

tasks_prompts_chain/tasks_prompts_chain.py

@@ -121,7 +121,7 @@ class TasksPromptsChain:
         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.
         
@@ -132,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
         """
@@ -184,14 +182,12 @@ class TasksPromptsChain:
                         response_content += delta
                         self._current_stream_buffer = response_content
                         self._format_current_stream()
-                        yield response_content
-                
-                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
+                        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 delta                
 
         except Exception as e:
             raise Exception(f"Error in prompt chain execution at prompt {i}: {str(e)}")
@@ -235,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.3.dist-info → tasks_prompts_chain-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tasks_prompts_chain
-Version: 0.0.3
+Version: 0.0.5
 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
@@ -59,8 +59,13 @@ from tasks_prompts_chain import TasksPromptsChain
 async def main():
     # Initialize the chain
     chain = TasksPromptsChain(
-        model="gpt-3.5-turbo",
-        api_key="your-api-key",
+        model_options={
+            "model": "gpt-3.5-turbo",
+            "api_key": "your-api-key",
+            "temperature": 0.1,
+            "max_tokens": 4120,
+            "stream": True
+        },
         final_result_placeholder="design_result"
     )
 
@@ -89,8 +94,43 @@ async def main():
 
 ## Advanced Usage
 
+### Using System Prompts
+
+```python
+chain = TasksPromptsChain(
+    model_options={
+            "model": "gpt-3.5-turbo",
+            "api_key": "your-api-key",
+            "temperature": 0.1,
+            "max_tokens": 4120,
+            "stream": True
+        },
+    final_result_placeholder="result",
+    system_prompt="You are a professional design expert specialized in luxury products",
+    system_apply_to_all_prompts=True
+)
+```
+
+### Custom API Endpoint
+
+```python
+chain = TasksPromptsChain(
+    model_options={
+            "model": "your-custom-model",
+            "api_key": "your-api-key",
+            "base_url": "https://your-custom-endpoint.com/v1",
+            "temperature": 0.1,
+            "max_tokens": 4120,
+            "stream": True
+        },
+    final_result_placeholder="result",
+)
+```
+
 ### 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("""
@@ -107,29 +147,13 @@ chain.template_output("""
 </result>
 """)
 ```
-
-### Using System Prompts
+then retrieves the final result within the template : 
 
 ```python
-chain = TasksPromptsChain(
-    model="gpt-3.5-turbo",
-    api_key="your-api-key",
-    final_result_placeholder="result",
-    system_prompt="You are a professional design expert specialized in luxury products",
-    system_apply_to_all_prompts=True
-)
+# print out the final result in the well formated template
+print(chain.get_final_result_within_template())
 ```
 
-### Custom API Endpoint
-
-```python
-chain = TasksPromptsChain(
-    model="gpt-3.5-turbo",
-    api_key="your-api-key",
-    final_result_placeholder="result",
-    base_url="https://your-custom-endpoint.com/v1"
-)
-```
 
 ## API Reference
 
@@ -152,6 +176,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.5.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=Ng1r8GvpAEKcOxMriX35lrmB74xHv2lZBMn1Mxs3AqE,9975
+tasks_prompts_chain-0.0.5.dist-info/METADATA,sha256=Cg_LG_ZN60YK-MPAkoIeblVwDVO7ATdmBRvbZKZQZgs,5778
+tasks_prompts_chain-0.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+tasks_prompts_chain-0.0.5.dist-info/licenses/LICENSE,sha256=WYmcYJG1QFgu1hfo7qrEkZ3Jhcz8NUWe6XUraZvlIFs,10172
+tasks_prompts_chain-0.0.5.dist-info/RECORD,,

tasks_prompts_chain-0.0.3.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=P6vJpuuwteXItmudhOnsiArojINeXB5lJnHgxTsXjCA,11816
-tasks_prompts_chain-0.0.3.dist-info/METADATA,sha256=MhEBZjJQ_7m8jH9RxxtEGSNn2ppEIR0DBoGIUmo25bA,4906
-tasks_prompts_chain-0.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-tasks_prompts_chain-0.0.3.dist-info/licenses/LICENSE,sha256=WYmcYJG1QFgu1hfo7qrEkZ3Jhcz8NUWe6XUraZvlIFs,10172
-tasks_prompts_chain-0.0.3.dist-info/RECORD,,