added gemini inference bot
This commit is contained in:
@@ -14,27 +14,30 @@ class StandaloneLLMTool(BaseTool):
|
||||
def get_functions(self):
|
||||
return [
|
||||
{
|
||||
"name": "call_external_llm",
|
||||
"description": "Call an external language model",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"prompt": {
|
||||
"type": "string",
|
||||
"description": "The prompt you are providing"
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "call_external_llm",
|
||||
"description": "Call an external language model",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"prompt": {
|
||||
"type": "string",
|
||||
"description": "The prompt you are providing"
|
||||
},
|
||||
"model": {
|
||||
"type": "string",
|
||||
"description": "The model to use for generating the detailed instructions. Use mini for most coding tasks, preview when needing sophisticated reasoning",
|
||||
"enum": ["o1-mini", "o1-preview"],
|
||||
"default": "o1-mini"
|
||||
},
|
||||
"max_tokens": {
|
||||
"type": "integer",
|
||||
"description": "The maximum number of tokens to use for generating the detailed instructions. Default is 16384.",
|
||||
}
|
||||
},
|
||||
"model": {
|
||||
"type": "string",
|
||||
"description": "The model to use for generating the detailed instructions. Use mini for most coding tasks, preview when needing sophisticated reasoning",
|
||||
"enum": ["o1-mini", "o1-preview"],
|
||||
"default": "o1-mini"
|
||||
},
|
||||
"max_tokens": {
|
||||
"type": "integer",
|
||||
"description": "The maximum number of tokens to use for generating the detailed instructions. Default is 16384.",
|
||||
}
|
||||
},
|
||||
"required": ["prompt"]
|
||||
"required": ["prompt"]
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user