truefoundry 0.10.3rc1__py3-none-any.whl → 0.10.4rc2__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/builder/builders/tfy_spark_buildpack/dockerfile_template.py

@@ -24,6 +24,9 @@ COPY ${requirements_path} ${requirements_destination_path}
 % if python_packages_install_command is not None:
 RUN ${package_manager_config_secret_mount} ${python_packages_install_command}
 % endif
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV IPYTHONDIR=/tmp/.ipython
+RUN groupadd --system --gid 1001 spark && useradd --system --uid 1001 --gid spark --no-create-home spark
 USER spark
 COPY . /app
 """

truefoundry/deploy/python_deploy_codegen.py

@@ -8,6 +8,7 @@ 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(
@@ -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,25 @@ 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):
+    original_repr_args = BaseModel.__repr_args__
+
+    def _patched_repr_args(self: BaseModel):
+        return [
+            (name, value)
+            for name, value in original_repr_args(self)
+            if name in self.__fields_set__
+        ]
+
+    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.3rc1.dist-info → truefoundry-0.10.4rc2.dist-info}/METADATA

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

{truefoundry-0.10.3rc1.dist-info → truefoundry-0.10.4rc2.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=CNL1633k84Mg4hUk2azc88TrQwRWEON0f2Y8V3SGuoM,7173
 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
@@ -66,7 +66,7 @@ truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.p
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=_fjqHKn80qKi68SAMMALge7_A6e1sTsQWichw8uoGIw,2025
 truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=f4l3fH21E2b8W3-JotMKc0AdPcCxV7LRPxxYJa7z_UQ,9134
 truefoundry/deploy/builder/builders/tfy_spark_buildpack/__init__.py,sha256=NEPlM6_vTVxp4ITa18B8DBbgYCn1q5d8be21lbgu5oY,2888
-truefoundry/deploy/builder/builders/tfy_spark_buildpack/dockerfile_template.py,sha256=czPBD29Q8Ua9NQNA-EOgx8hrafA22fiE14ZNB6wyLVc,5826
+truefoundry/deploy/builder/builders/tfy_spark_buildpack/dockerfile_template.py,sha256=dAmWgAHcNdBLSedpz18r5r7n9hdXKV-PB7yFPMdaoww,5990
 truefoundry/deploy/builder/builders/tfy_spark_buildpack/tfy_execute_notebook.py,sha256=PKJWNlW76dRw_XtFkRnqpzwBGuXU4w6_kIJkgoBmZ5Q,1975
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/cli/commands/__init__.py,sha256=qv818jxqSAygJ3h-6Ul8t-5VOgR_UrSgsVtNCl3e5G0,1408
@@ -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.3rc1.dist-info/METADATA,sha256=nfiSHIUJ48mtIr9hXuPV_zCdJ-e5JNKXQ5_yC2GHnGk,2508
-truefoundry-0.10.3rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.10.3rc1.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.10.3rc1.dist-info/RECORD,,
+truefoundry-0.10.4rc2.dist-info/METADATA,sha256=SJpxZlTvIDJc5NOKQZdbj-K4z1tv0Dqct9vdAwYuKYw,2508
+truefoundry-0.10.4rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.10.4rc2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.10.4rc2.dist-info/RECORD,,