Henri Bonamy commited on
Commit
e6cb515
·
1 Parent(s): 8737058

small prompt change, printing job url before stream instead of id

Browse files
agent/prompts/system_prompt.yaml CHANGED
@@ -87,6 +87,7 @@ system_prompt: |
87
  - Never assume a library is available - check documentation first
88
  - Before processing any dataset: inspect its actual structure first using the mcp__hf-mcp-server__hub_repo_details tool. Never assume column names: verify them beforehand.
89
  - Follow ML best practices: proper train/val/test splits, reproducibility, evaluation metrics
 
90
  - For training tasks, consider compute requirements and suggest appropriate hardware
91
  - Never expose or log API keys, tokens, or secrets
92
 
 
87
  - Never assume a library is available - check documentation first
88
  - Before processing any dataset: inspect its actual structure first using the mcp__hf-mcp-server__hub_repo_details tool. Never assume column names: verify them beforehand.
89
  - Follow ML best practices: proper train/val/test splits, reproducibility, evaluation metrics
90
+ - Unless absolutely necessary, don't ask user for action. This does not apply to follow-up questions you have.
91
  - For training tasks, consider compute requirements and suggest appropriate hardware
92
  - Never expose or log API keys, tokens, or secrets
93
 
agent/tools/jobs_tool.py CHANGED
@@ -448,7 +448,7 @@ To inspect, call this tool with `{{"operation": "inspect", "args": {{"job_id": "
448
  return {"formatted": response, "totalResults": 1, "resultsShared": 1}
449
 
450
  # Not detached - wait for completion and stream logs
451
- print(f"Job started: {job.id}")
452
  print("Streaming logs...\n---\n")
453
 
454
  final_status, all_logs = await self._wait_for_job_completion(
@@ -521,7 +521,7 @@ To check logs, call this tool with `{{"operation": "logs", "args": {{"job_id": "
521
  return {"formatted": response, "totalResults": 1, "resultsShared": 1}
522
 
523
  # Not detached - wait for completion and stream logs
524
- print(f"UV Job started: {job.id}")
525
  print("Streaming logs...\n---\n")
526
 
527
  final_status, all_logs = await self._wait_for_job_completion(
 
448
  return {"formatted": response, "totalResults": 1, "resultsShared": 1}
449
 
450
  # Not detached - wait for completion and stream logs
451
+ print(f"Job started: {job.url}")
452
  print("Streaming logs...\n---\n")
453
 
454
  final_status, all_logs = await self._wait_for_job_completion(
 
521
  return {"formatted": response, "totalResults": 1, "resultsShared": 1}
522
 
523
  # Not detached - wait for completion and stream logs
524
+ print(f"UV Job started: {job.url}")
525
  print("Streaming logs...\n---\n")
526
 
527
  final_status, all_logs = await self._wait_for_job_completion(