Harshil Patel commited on
Commit
8a91ce3
·
1 Parent(s): 98d31e7

add more gemini agents

Browse files
Files changed (1) hide show
  1. default_tools/agent_creater_tool.py +54 -4
default_tools/agent_creater_tool.py CHANGED
@@ -18,7 +18,7 @@ class AgentCreator():
18
  },
19
  "base_model": {
20
  "type": "string",
21
- "description": "A base model from which the new agent mode is to be created. Available models are: llama3.2, mistral, gemini-2.0-flash"
22
  },
23
  "system_prompt": {
24
  "type": "string",
@@ -44,10 +44,60 @@ class AgentCreator():
44
  "create_cost": 20,
45
  "invoke_cost": 50,
46
  },
 
 
 
 
 
 
 
 
 
 
47
  "gemini-2.0-flash": {
48
- "description": "40 Billion parameter model",
49
- "create_cost": 30,
50
- "invoke_cost": 60,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  }
52
  }
53
  }
 
18
  },
19
  "base_model": {
20
  "type": "string",
21
+ "description": "A base model from which the new agent mode is to be created. Available models are: llama3.2, mistral, gemini-2.5-flash-preview-04-17, gemini-2.5-pro-preview-03-25, gemini-2.0-flash, gemini-2.0-flash-lite, gemini-1.5-flash, gemini-1.5-flash-8b, gemini-1.5-pro, and gemini-2.0-flash-live-001"
22
  },
23
  "system_prompt": {
24
  "type": "string",
 
44
  "create_cost": 20,
45
  "invoke_cost": 50,
46
  },
47
+ "gemini-2.5-flash-preview-04-17": {
48
+ "description": "Adaptive thinking, cost efficiency",
49
+ "create_cost": 20,
50
+ "invoke_cost": 50
51
+ },
52
+ "gemini-2.5-pro-preview-03-25": {
53
+ "description": "Enhanced thinking and reasoning, multimodal understanding, advanced coding, and more",
54
+ "create_cost": 20,
55
+ "invoke_cost": 50
56
+ },
57
  "gemini-2.0-flash": {
58
+ "description": "Next generation features, speed, thinking, realtime streaming, and multimodal generation",
59
+ "create_cost": 20,
60
+ "invoke_cost": 50
61
+ },
62
+ "gemini-2.0-flash-lite": {
63
+ "description": "Cost efficiency and low latency",
64
+ "create_cost": 20,
65
+ "invoke_cost": 50
66
+ },
67
+ "gemini-1.5-flash": {
68
+ "description": "Fast and versatile performance across a diverse variety of tasks",
69
+ "create_cost": 20,
70
+ "invoke_cost": 50
71
+ },
72
+ "gemini-1.5-flash-8b": {
73
+ "description": "High volume and lower intelligence tasks",
74
+ "create_cost": 20,
75
+ "invoke_cost": 50
76
+ },
77
+ "gemini-1.5-pro": {
78
+ "description": "Complex reasoning tasks requiring more intelligence",
79
+ "create_cost": 20,
80
+ "invoke_cost": 50
81
+ },
82
+ # "gemini-embedding-exp": {
83
+ # "description": "Measuring the relatedness of text strings",
84
+ # "create_cost": 20,
85
+ # "invoke_cost": 50
86
+ # },
87
+ # "imagen-3.0-generate-002": {
88
+ # "description": "Our most advanced image generation model",
89
+ # "create_cost": 20,
90
+ # "invoke_cost": 50
91
+ # },
92
+ # "veo-2.0-generate-001": {
93
+ # "description": "High quality video generation",
94
+ # "create_cost": 20,
95
+ # "invoke_cost": 50
96
+ # },
97
+ "gemini-2.0-flash-live-001": {
98
+ "description": "Low-latency bidirectional voice and video interactions",
99
+ "create_cost": 20,
100
+ "invoke_cost": 50
101
  }
102
  }
103
  }