truefoundry 0.10.4rc1__py3-none-any.whl → 0.10.4rc3__py3-none-any.whl

truefoundry/_ask/client.py

@@ -94,7 +94,6 @@ class AskClient:
         """Generate authorization headers for connecting to the SSE server."""
         return {
             "Authorization": f"Bearer {self.token}",
-            "X-TFY-Cluster-Id": self.cluster,
             "X-TFY-Session-Id": self._session_id,
         }
 
@@ -317,6 +316,10 @@ class AskClient:
                     turn += 1
                     message = response.choices[0].message
 
+                    if not message.content and not message.tool_calls:
+                        self._log_message("No assistant response. Try again.", log=True)
+                        break
+
                     if message.content:
                         self._append_message(
                             ChatCompletionAssistantMessageParam(
@@ -327,8 +330,7 @@ class AskClient:
                     if message.tool_calls:
                         await self._handle_tool_calls(message, spinner)
 
-                    if not message.content and not message.tool_calls:
-                        self._log_message("No assistant response.")
+                    if message.content and not message.tool_calls:
                         break
                 except Exception as e:
                     self._log_message(f"OpenAI call failed: {e}", log=True)
@@ -402,10 +404,16 @@ class AskClient:
 
     async def chat_loop(self):
         """Interactive loop: accepts user queries and returns responses until interrupted or 'exit' is typed."""
+        self._append_message(
+            ChatCompletionUserMessageParam(
+                role="user",
+                content=f"Selected cluster: {self.cluster}",
+            )
+        )
         self._append_message(
             ChatCompletionAssistantMessageParam(
                 role="assistant",
-                content="Hello! How can I help you with your Kubernetes cluster?",
+                content="Hello! How can I help you with this Kubernetes cluster?",
             )
         )
 

truefoundry/deploy/python_deploy_codegen.py

@@ -2,12 +2,13 @@ import ast
 import io
 import json
 import re
-from typing import Dict, List, Optional
+from typing import Any, Dict, List, Optional
 
 from rich.console import Console
 from rich.pretty import pprint
 
 from truefoundry.deploy import Application, LocalSource
+from truefoundry.pydantic_v1 import BaseModel
 
 
 def generate_deployment_code(
@@ -16,7 +17,7 @@ def generate_deployment_code(
     spec_repr: str,
     workspace_fqn: str,
 ):
-    symbols = ",".join(symbols_to_import)
+    symbols = ", ".join(symbols_to_import)
     application_type = application_type.replace(" ", "").replace("-", "_")
     code = f"""\
 import logging
@@ -71,7 +72,7 @@ def remove_none_type_fields(code):
 
 def remove_type_field(code):
     lines = code.split("\n")
-    new_lines = [re.sub(r'^[ \t]*type=[\'"][^"]*[\'"],', "", line) for line in lines]
+    new_lines = [re.sub(r'^[ \t]*type=[\'"][^"]*[\'"],?', "", line) for line in lines]
     return "\n".join(new_lines)
 
 
@@ -102,7 +103,7 @@ def add_local_source_comment(code):
     return "\n".join(new_lines)
 
 
-def convert_deployment_config_to_python(workspace_fqn: str, application_spec: dict):
+def _convert_deployment_config_to_python(workspace_fqn: str, application_spec: dict):
     """
     Convert a deployment config to a python file that can be used to deploy to a workspace
     """
@@ -142,6 +143,42 @@ def convert_deployment_config_to_python(workspace_fqn: str, application_spec: di
     return generated_code
 
 
+def convert_deployment_config_to_python(
+    workspace_fqn: str,
+    application_spec: Dict[str, Any],
+    exclude_unset: bool = False,
+    exclude_defaults: bool = False,
+):
+    original_repr_args = BaseModel.__repr_args__
+
+    def _patched_repr_args(self: BaseModel):
+        _missing = object()
+        pairs = []
+        for name, value in original_repr_args(self):
+            if name is not None:
+                model_field = self.__fields__.get(name)
+                if model_field is None:
+                    continue
+                if exclude_unset and name not in self.__fields_set__:
+                    continue
+                if (
+                    exclude_defaults
+                    and not getattr(model_field, "required", True)
+                    and getattr(model_field, "default", _missing) == value
+                ):
+                    continue
+            pairs.append((name, value))
+        return pairs
+
+    try:
+        BaseModel.__repr_args__ = _patched_repr_args
+        return _convert_deployment_config_to_python(
+            workspace_fqn=workspace_fqn, application_spec=application_spec
+        )
+    finally:
+        BaseModel.__repr_args__ = original_repr_args
+
+
 def generate_python_snippet_for_trigger_job(
     application_fqn: str, command: Optional[str], params: Optional[Dict[str, str]]
 ):

{truefoundry-0.10.4rc1.dist-info → truefoundry-0.10.4rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.10.4rc1
+Version: 0.10.4rc3
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1

{truefoundry-0.10.4rc1.dist-info → truefoundry-0.10.4rc3.dist-info}/RECORD

@@ -5,7 +5,7 @@ truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,34
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
 truefoundry/_ask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/_ask/cli.py,sha256=zPaDvxhX2dITmPTtut2Iu6WAIaizrwR-U_dDZ6xv2io,5814
-truefoundry/_ask/client.py,sha256=4vWO04jWbSF0XD3q8DwXjvL4HW-WBg7nsQ0DydNwHmM,18479
+truefoundry/_ask/client.py,sha256=QWQRiDwmtIlLaZsyGcLZaQstYFzpmJeCRdATMapjL-8,18740
 truefoundry/_ask/llm_utils.py,sha256=ayjz7JtVu142lrm8t0cVoxLxUpx76b71y8R62z_WurY,13537
 truefoundry/autodeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/cli.py,sha256=9ZxKu_MGIpraMzaW4ZyuQZhlKIQYE3biBrBV4S1h6Fo,14167
@@ -53,7 +53,7 @@ truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,1
 truefoundry/common/utils.py,sha256=j3QP0uOsaGD_VmDDR68JTwoYE1okkAq6OqpVkzVf48Q,6424
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=2GNbI8IGJBotz_IKaqQ-DWYWZn_pSu7lN7aId15Gk7Q,2799
-truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
+truefoundry/deploy/python_deploy_codegen.py,sha256=X6cSGQ9_9GxrgIlTLvBWMDz9QnU7hrxieMIutNJe_ng,7784
 truefoundry/deploy/_autogen/models.py,sha256=xt-DuaRDx5jeRwyGoQH2yyPZAep9Q2MHFW9XBuRzG8E,73161
 truefoundry/deploy/builder/__init__.py,sha256=kgvlkVkiWpMVdim81tIeLrdoACqrFDgwCqHdQVsCsMo,4988
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
@@ -381,7 +381,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.10.4rc1.dist-info/METADATA,sha256=925B7ewlbtV30Q3WkQNhKyq52X2Opd83SSRqsFZpYjY,2508
-truefoundry-0.10.4rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.10.4rc1.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.10.4rc1.dist-info/RECORD,,
+truefoundry-0.10.4rc3.dist-info/METADATA,sha256=DvsgKrey42e5PHmRKl107ZHQeOO-uqsRX-OG7tibhiI,2508
+truefoundry-0.10.4rc3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.10.4rc3.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.10.4rc3.dist-info/RECORD,,