evapilotno17 commited on
Commit
68d0bb2
·
verified ·
1 Parent(s): ac5aeb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -12
app.py CHANGED
@@ -40,6 +40,33 @@ final_answer = FinalAnswerTool()
40
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
41
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
42
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
43
  model = HfApiModel(
44
  max_tokens=2096,
45
  temperature=0.5,
@@ -57,18 +84,7 @@ with open("prompts.yaml", 'r') as stream:
57
 
58
  agent = CodeAgent(
59
  model=model,
60
- tools=[
61
- final_answer,
62
- get_contest_data,
63
- get_rating_data,
64
- count_of_rated_matches,
65
- get_user_info,
66
- get_handle,
67
- solve_cutoff,
68
- get_all_rated_users,
69
- filter_rated_users,
70
- intersect,
71
- ], ## add your tools here (don't remove final answer)
72
  max_steps=6,
73
  verbosity_level=1,
74
  grammar=None,
 
40
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
41
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
42
 
43
+ tools = [
44
+ final_answer,
45
+ get_contest_data,
46
+ get_rating_data,
47
+ count_of_rated_matches,
48
+ get_user_info,
49
+ get_handle,
50
+ solve_cutoff,
51
+ get_all_rated_users,
52
+ filter_rated_users,
53
+ intersect,
54
+ get_user_info,
55
+ get_contribution,
56
+ get_lastOnlineTimeSeconds,
57
+ get_organization,
58
+ get_rating,
59
+ get_friendOfCount,
60
+ get_titlePhoto,
61
+ get_rank,
62
+ get_handle,
63
+ get_maxRating,
64
+ get_avatar,
65
+ get_registrationTimeSeconds,
66
+ get_maxRank
67
+ ]
68
+
69
+
70
  model = HfApiModel(
71
  max_tokens=2096,
72
  temperature=0.5,
 
84
 
85
  agent = CodeAgent(
86
  model=model,
87
+ tools=tools, ## add your tools here (don't remove final answer)
 
 
 
 
 
 
 
 
 
 
 
88
  max_steps=6,
89
  verbosity_level=1,
90
  grammar=None,