yetessam commited on
Commit
4c9915a
·
verified ·
1 Parent(s): 8252cd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -13,7 +13,7 @@ from smolagents import CodeAgent, InferenceClientModel
13
  from checks.status_check import resolve_endpoint, is_endpoint_healthy
14
 
15
  from tools.tools import load_tools
16
- from prompts.prompts import load_prompts
17
 
18
  def initialize_agent(endpoint_uri: str):
19
 
@@ -60,9 +60,16 @@ def main():
60
  # If the end point is ready, don't show it.
61
 
62
 
63
- agent_type = """ deepseek-ai/DeepSeek-R1-Distill-Qwen-32B` text generation model """
64
  compute = "GCP · Nvidia L4 · 4x GPUs · 96 GB"
65
-
 
 
 
 
 
 
 
66
  ui = ContentAgentUI(
67
  endpoint_uri=uri,
68
  is_healthy=ok,
 
13
  from checks.status_check import resolve_endpoint, is_endpoint_healthy
14
 
15
  from tools.tools import load_tools
16
+ from prompts.prompts import load_prompts, print_default_prompt_template
17
 
18
  def initialize_agent(endpoint_uri: str):
19
 
 
60
  # If the end point is ready, don't show it.
61
 
62
 
63
+ agent_type = 'deepseek-ai/DeepSeek-R1-Distill-Qwen-32B'
64
  compute = "GCP · Nvidia L4 · 4x GPUs · 96 GB"
65
+
66
+ print_default_prompt_templates(
67
+ base_url=uri,
68
+ model_id=agent_type,
69
+ api_key=None,);
70
+
71
+ exit 0
72
+
73
  ui = ContentAgentUI(
74
  endpoint_uri=uri,
75
  is_healthy=ok,