waldiez 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.2"
+__version__ = "0.1.4"

waldiez/exporting/flow/flow.py

@@ -80,19 +80,22 @@ def export_flow(
         "import os",
         "import sqlite3",
     }
-    other_imports: Set[str] = {
+    common_imports: Set[str] = {
         "from autogen import Agent",
         "from autogen import ConversableAgent",
         "from autogen import ChatResult",
         "from autogen import runtime_logging",
     }
+    local_imports: Set[str] = {
+        "from waldiez_api_keys import get_model_api_key",
+    }
     skill_imports, _ = export_skills(
         skills=all_skills,
         skill_names=skill_names,
         output_dir=output_dir,
     )
     if len(waldiez.chats) > 1:
-        other_imports.add("from autogen import initiate_chats")
+        common_imports.add("from autogen import initiate_chats")
     for agent in all_agents:
         agent_string, after_agent, agent_imports = export_agent(
             agent=agent,
@@ -103,7 +106,7 @@ def export_flow(
             all_skills=all_skills,
             group_chat_members=waldiez.flow.get_group_chat_members(agent.id),
         )
-        other_imports.update(agent_imports)
+        common_imports.update(agent_imports)
         if after_agent:
             skipped_agent_strings += after_agent
         if agent.agent_type == "manager":
@@ -123,11 +126,13 @@ def export_flow(
         all_models=all_models,
         model_names=model_names,
         notebook=notebook,
+        output_dir=output_dir,
     )
     all_imports_string = get_imports_string(
-        imports=other_imports,
+        imports=common_imports,
         builtin_imports=builtin_imports,
         skill_imports=skill_imports,
+        local_imports=local_imports,
     )
     return _combine_strings(
         waldiez=waldiez,

waldiez/exporting/models/__init__.py

@@ -6,7 +6,8 @@ export_models
     Get the string representations of the LLM configs.
 """
 
-from typing import Dict, List
+from pathlib import Path
+from typing import Dict, List, Optional
 
 from waldiez.models import WaldiezModel
 
@@ -17,6 +18,7 @@ def export_models(
     all_models: List[WaldiezModel],
     model_names: Dict[str, str],
     notebook: bool,
+    output_dir: Optional[Path] = None,
 ) -> str:
     """Get the string representations of the LLM configs.
 
@@ -28,7 +30,8 @@ def export_models(
         A mapping of model ids to model names.
     notebook : bool
         Whether to export the string for a jupyter notebook.
-
+    output_dir : Optional[Path]
+        The output directory to write the api keys.
     Returns
     -------
     str
@@ -38,7 +41,8 @@ def export_models(
     -------
     ```python
     >>> from waldiez.models import WaldiezModel, WaldiezModelData
-    >>> model = WaldiezModel(
+    >>> api_key = "1234567890"
+    ... model = WaldiezModel(
     ...     id="wm-1",
     ...     name="llama3.1"  ,
     ...     description="A model for llamas :P.",
@@ -46,7 +50,7 @@ def export_models(
     ...     requirements=[],
     ...     data=WaldiezModelData(
     ...         base_url="https://example.com/v1",
-    ...         api_key="1234567890",
+    ...         api_key=api_key,
     ...         api_type="openai",
     ...         temperature=0.5,
     ...         price={
@@ -61,9 +65,9 @@ def export_models(
     # # Models
     llama3_1_llm_config = {
         "config_list": [{
-            "model": "llama3.1",
+            "model": "llama3_1",
             "base_url": "https://example.com/v1",
-            "api_key": "1234567890",
+            "api_key": get_model_api_key("llama3_1"),
             "api_type": "openai",
             "temperature": 0.5,
             "price": [0.0001, 0.0002],
@@ -72,26 +76,22 @@ def export_models(
     ```
     """
     content = get_comment("models", notebook) + "\n"
-    if len(all_models) == 1:
-        only_model = all_models[0]
-        model_name = model_names[only_model.id]
-        llm_config = only_model.get_llm_config()
+    for model in all_models:
+        model_name = model_names[model.id]
+        llm_config = model.get_llm_config()
+        llm_config["api_key"] = f'get_model_api_key("{model_name}")'
         model_dict_str = get_object_string(llm_config, tabs=2)
+        model_dict_str = model_dict_str.replace(
+            f'"get_model_api_key("{model_name}")"',
+            f'get_model_api_key("{model_name}")',
+        )
         content += f"{model_name}_llm_config = " + "{\n"
         content += '    "config_list": [\n'
         content += f"        {model_dict_str}\n"
         content += "    ]\n"
         content += "}\n"
-    else:
-        for model in all_models:
-            model_name = model_names[model.id]
-            llm_config = model.get_llm_config()
-            model_dict_str = get_object_string(llm_config, tabs=2)
-            content += f"{model_name}_llm_config = " + "{\n"
-            content += '    "config_list": [\n'
-            content += f"        {model_dict_str}\n"
-            content += "    ]\n"
-            content += "}\n"
+    if output_dir:
+        write_api_keys(all_models, model_names, output_dir)
     return content
 
 
@@ -191,3 +191,52 @@ def export_agent_models(
     content += "    ]\n"
     content += "}\n"
     return content
+
+
+def write_api_keys(
+    all_models: List[WaldiezModel],
+    model_names: Dict[str, str],
+    output_dir: Path,
+) -> None:
+    """Write the api keys to a separate file.
+
+    Parameters
+    ----------
+    all_models : List[WaldiezModel]
+        All the models in the flow.
+    model_names : Dict[str, str]
+        A mapping of model ids to model names.
+    output_dir : Path
+        The output directory to write the api keys.
+    """
+    # example call: llama3_1_api_key = get_model_api_key("llama3_1")
+    api_keys_content = '''
+"""API keys for the models."""
+
+__ALL_MODEL_API_KEYS__ = {
+'''
+    for model in all_models:
+        model_name = model_names[model.id]
+        api_keys_content += f'    "{model_name}": "{model.api_key}",\n'
+    api_keys_content += "}\n"
+
+    api_keys_content += '''
+
+def get_model_api_key(model_name: str) -> str:
+    """Get the api key for the model.
+
+    Parameters
+    ----------
+    model_name : str
+        The name of the model.
+
+    Returns
+    -------
+    str
+        The api key for the model.
+    """
+    return __ALL_MODEL_API_KEYS__.get(model_name, "")
+'''
+
+    with open(output_dir / "waldiez_api_keys.py", "w", encoding="utf-8") as f:
+        f.write(api_keys_content)

waldiez/exporting/utils/importing.py

@@ -65,7 +65,7 @@ def get_imports_string(
     skill_imports: Set[str],
     typing_imports: Optional[Set[str]] = None,
     builtin_imports: Optional[Set[str]] = None,
-    other_imports: Optional[Set[str]] = None,
+    local_imports: Optional[Set[str]] = None,
 ) -> str:
     """Get the imports.
 
@@ -79,8 +79,8 @@ def get_imports_string(
         The typing imports, by default None.
     builtin_imports : Set[str], optional
         The builtin imports, by default None.
-    other_imports : Set[str], optional
-        Other third party imports, by default None.
+    local_imports : Set[str], optional
+        The local imports (like for getting model api keys), by default None.
 
     Returns
     -------
@@ -107,10 +107,12 @@ def get_imports_string(
         typing_imports = DEFAULT_TYPING_IMPORTS
     if not builtin_imports:
         builtin_imports = set()
-    if not other_imports:
-        other_imports = set()
+    if not local_imports:
+        local_imports = set()
     string = _get_builtin_imports_string(builtin_imports, typing_imports)
-    string += _get_third_party_imports_string(imports, other_imports)
+    string += _get_third_party_imports_string(imports)
+    if local_imports:
+        string += "\n\n" + "\n".join(sorted(local_imports)) + "\n"
     string += _get_skill_imports_string(skill_imports)
     string = "\n\n".join([line for line in string.split("\n\n") if line])
     while not string.endswith("\n\n"):
@@ -234,13 +236,9 @@ def _get_autogen_imports(
     return autogen_imports, autogen_dot_imports, remaining_imports
 
 
-def _get_third_party_imports_string(
-    imports: Set[str],
-    other_imports: Set[str],
-) -> str:
+def _get_third_party_imports_string(imports: Set[str]) -> str:
     """Get the third party imports."""
     autogen_imports, autogen_dot_imports, rest = _get_autogen_imports(imports)
-    rest.update(other_imports)
     plain_imports, from_imports_dict = _prepare_imports(
         autogen_imports=autogen_imports,
         autogen_dot_imports=autogen_dot_imports,

waldiez/models/agents/agent/termination_message.py

@@ -136,7 +136,8 @@ class WaldiezAgentTerminationMessage(WaldiezBase):
             )
         if self.criterion == "ending":
             self._string = (
-                'lambda x: any(x.get("content", "").endswith(keyword) '
+                'lambda x: any(x.get("content", "") and '
+                'x.get("content", "").endswith(keyword) '
                 f"for keyword in [{keywords_str}])"
             )
         if self.criterion == "exact":

{waldiez-0.1.2.dist-info → waldiez-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.2
+Version: 0.1.4
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/py/
 Project-URL: repository, https://github.com/waldiez/py.git
@@ -177,4 +177,4 @@ io_stream.close()
 
 ## License
 
-This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+This project is licensed under the MIT License - see the [LICENSE](https://github.com/waldiez/py/blob/main/LICENSE) file for details.

{waldiez-0.1.2.dist-info → waldiez-0.1.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 waldiez/__init__.py,sha256=pX6lHj8fNlfqCpHEfF6MNQBCR6qRjFYjwBSOCoxrlwk,313
 waldiez/__main__.py,sha256=9xR-F2ohZcRPDG6KrM7cJpXciKX-u6WdL221ckyJ04k,112
-waldiez/_version.py,sha256=hiO5XGIMf-JKIDO7TZrWy6g1Iq-nh3n_W-WL3Cll4ZM,62
+waldiez/_version.py,sha256=N-ZsihH4RZovpgrN6mkwtpRExfYJUp8a4NSA9bZB5WA,62
 waldiez/cli.py,sha256=vIQDqs9XBOlDpshq7rNgvoTcnnkeZrsXa_51gMWs6H4,4690
 waldiez/exporter.py,sha256=iKe-l_Me8NRWsXHIdBcrOrnLT9XIyp4iYi4HLuuj2jA,9342
 waldiez/io_stream.py,sha256=YczhIw0PKEsxCSY6tAqElCqOUh8b-aEkSIGC1JjXxAs,5877
@@ -28,12 +28,12 @@ waldiez/exporting/chats/helpers.py,sha256=LmbOKXxMMSf2D1up8CNxxitUDvZ-Z_8NNxJFcP
 waldiez/exporting/chats/nested.py,sha256=39qodz8Lct2SDjIjEHSxNUls9T4UtfLBEeiBcmqydbA,8254
 waldiez/exporting/flow/__init__.py,sha256=WhdPrjXQAcihrS1KUtPNgbx0y1tqD5HtGykzpEAcsBM,98
 waldiez/exporting/flow/def_main.py,sha256=YpdhqO4iFng3r7if69ZPMJAibPVNDqnrOrWvGw7CJq8,1052
-waldiez/exporting/flow/flow.py,sha256=AGISfMOicdee4zikvuega8ZnHceQpdfCphDBgIE5_94,6038
-waldiez/exporting/models/__init__.py,sha256=qJvP0CIrgLPM2pnIoXM26eyUdh7ub-CiNNLJwXEOAwQ,5829
+waldiez/exporting/flow/flow.py,sha256=x2AlXr7aJJfrPMeuXQYfT9sG5GycMwf0kRkovZWKGag,6206
+waldiez/exporting/models/__init__.py,sha256=_Yt6sBAyRrN3LSAg4Nrh4dP7pmtIzGzWu4G1AutxK-Q,7078
 waldiez/exporting/skills/__init__.py,sha256=oIA9f5ABtYSbS0kMY_4ovU3-m6meVk5blEu_xViZsRU,3536
 waldiez/exporting/utils/__init__.py,sha256=tP1V4g9-MyARlfOEL_1YWMJNW7UivUrrukq7DIwdq6k,1018
 waldiez/exporting/utils/comments.py,sha256=X9j8w48rh3DfFDjiMverU9DBSuE9yuMMbbStxBbN1sE,3190
-waldiez/exporting/utils/importing.py,sha256=M7vBcW9gPe3hCaTX2Bsv-mz36DAiCe4R-3mLrMWGY-Y,8616
+waldiez/exporting/utils/importing.py,sha256=dA4HCQ-OxmodUjovgXyLI9IwNvLwbY67P41969XoZ7g,8649
 waldiez/exporting/utils/logging_utils.py,sha256=uoV6O23lfB5ztOYEZiYu8Mn-2xEUwp_Qx404Mr62i7M,5822
 waldiez/exporting/utils/method_utils.py,sha256=7-RUMTylNM2W0iM1bPX2_Gn3553XZSl2s2VGEijxNp4,891
 waldiez/exporting/utils/naming.py,sha256=VdoVODQduhXIs9hQFWUVEVqTaSyNDt7rkECsuIgXYwI,3196
@@ -50,7 +50,7 @@ waldiez/models/agents/agent/code_execution.py,sha256=kgL3pbEEyuMvJid0sIbfe4os7SW
 waldiez/models/agents/agent/linked_skill.py,sha256=8RHWHkHXqFuv7lEe1PuQoK1hTO3kBQ7ILKm9kCEWqNs,667
 waldiez/models/agents/agent/nested_chat.py,sha256=DAF3TCbTwyDvg6PGbeETtBLCQ4Xz5Saw5a4Xi-jr6jA,1929
 waldiez/models/agents/agent/teachability.py,sha256=IIR4LY9dwx3T7Ok17RYN2g6zGiga2gGizGteaeI3eGs,1703
-waldiez/models/agents/agent/termination_message.py,sha256=lZskWdFCHPNldTwy080rxCQNaEfyR08vzaKWCG00-Mw,5630
+waldiez/models/agents/agent/termination_message.py,sha256=aGsIdwGs42BxDktkxoqHGEWM1wzhXr-zh_MlETsbycA,5674
 waldiez/models/agents/assistant/__init__.py,sha256=Zlf-4EI9HXl-LrqGosL7UucoyqIl74GZzohZlRLx2QI,175
 waldiez/models/agents/assistant/assistant.py,sha256=OFh5nr8HrOLg2KXnwzFUU7_JBWrYfM7lRtTRQT1E_gI,1095
 waldiez/models/agents/assistant/assistant_data.py,sha256=VGdF1IZBBqBvZwe6BP2s4LinmkQ2IjesaDvs9Me9iHY,823
@@ -88,8 +88,8 @@ waldiez/stream/__init__.py,sha256=6qst5j2iXmV-wDTnhgCna3llqUfJ6tkR0HBq7Vx9x-Q,19
 waldiez/stream/consumer.py,sha256=VQDJEomYpGjmAtOoFBJrpCyoYzwykJUIOrdiLn08Uj4,4049
 waldiez/stream/provider.py,sha256=JqnnR1yAimfK4HxW4gr4HIoJ15Ly71kXn75mv7FZBmw,9478
 waldiez/stream/server.py,sha256=Ya4AKP5Wr_xAsbG3Q-qShqHmfIyamj9LGXizJsL-N4Y,11815
-waldiez-0.1.2.dist-info/METADATA,sha256=CoM_iWzFesJK4x_hhyAojVpt9joTUZMFiwRyxRoemoE,6745
-waldiez-0.1.2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-waldiez-0.1.2.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
-waldiez-0.1.2.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
-waldiez-0.1.2.dist-info/RECORD,,
+waldiez-0.1.4.dist-info/METADATA,sha256=ppqb9brA_gfW5p5aiiMNCgH9AS02k4gf1OVoyZLQXo0,6785
+waldiez-0.1.4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+waldiez-0.1.4.dist-info/entry_points.txt,sha256=5Po4yQXPa_QEdtTevpEBgr3rGoIvDMeQuJR2zqwBLBo,45
+waldiez-0.1.4.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
+waldiez-0.1.4.dist-info/RECORD,,