Skip to content

Commit 2cac6d1

Browse files
kevin2016张可成
andauthored
Add DM-Cito-8B (ShishirPatil#1017)
This PR adds model handler for DM-Cito-8B. The url of the model api is:https://rapidapi.com/kechengkevin/api/dm-cito-8b --------- Co-authored-by: 张可成 <[email protected]>
1 parent e6cafbc commit 2cac6d1

File tree

6 files changed

+33
-3
lines changed

6 files changed

+33
-3
lines changed

berkeley-function-call-leaderboard/.env.example

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,10 @@ COHERE_API_KEY=
1212
GROK_API_KEY=xai-XXXXXX
1313
GOGOAGENT_API_KEY=
1414
WRITER_API_KEY=
15-
MINING_API_KEY="sk-XXXXXX"
15+
MINING_BASE_URL=XXXXXX
16+
MINING_API_KEY=sk-XXXXXX
17+
DMCITO_BASE_URL=XXXXXX
18+
DMCITO_API_KEY=sk-XXXXXX
1619

1720
# We use Vertex AI to inference Google Gemini models
1821
VERTEX_AI_PROJECT_ID=

berkeley-function-call-leaderboard/SUPPORTED_MODELS.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ For model names containing `{...}`, multiple versions are available. For example
119119
| xLAM-2-70b-fc-r | Function Calling | Self-hosted 💻 | Salesforce/Llama-xLAM-2-70b-fc-r |
120120
| xLAM-2-8b-fc-r | Function Calling | Self-hosted 💻 | Salesforce/Llama-xLAM-2-8b-fc-r |
121121
| yi-large | Function Calling | 01.AI | yi-large-fc |
122-
122+
| DM-Cito-8B | Prompt | Mininglamp | DM-Cito-8B |
123123
---
124124

125125
## Additional Requirements for Certain Models

berkeley-function-call-leaderboard/bfcl/constants/model_config.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
from bfcl.model_handler.local_inference.salesforce_qwen import SalesforceQwenHandler
4646
from bfcl.model_handler.local_inference.think_agent import ThinkAgentHandler
4747
from bfcl.model_handler.api_inference.qwq import QwenAPIHandler
48+
from bfcl.model_handler.api_inference.dm_cito import DMCitoHandler
4849

4950
# -----------------------------------------------------------------------------
5051
# A mapping of model identifiers to their respective model configurations.
@@ -838,6 +839,18 @@ class ModelConfig:
838839
is_fc_model=False,
839840
underscore_to_dot=False,
840841
),
842+
"DM-Cito-8B": ModelConfig(
843+
model_name="DM-Cito-8B",
844+
display_name="DM-Cito-8B (Prompt)",
845+
url="https://www.mininglamp.com/",
846+
org="Mininglamp",
847+
license="Proprietary",
848+
model_handler=DMCitoHandler,
849+
input_price=None,
850+
output_price=None,
851+
is_fc_model=False,
852+
underscore_to_dot=False,
853+
),
841854
}
842855

843856

berkeley-function-call-leaderboard/bfcl/constants/supported_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@
7676
"grok-3-mini-beta",
7777
"QwQ-32B",
7878
"xiaoming-14B",
79+
"DM-Cito-8B",
7980
"deepseek-ai/DeepSeek-R1",
8081
"google/gemma-3-1b-it",
8182
"google/gemma-3-4b-it",
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
import os
2+
from bfcl.model_handler.model_style import ModelStyle
3+
from openai import OpenAI
4+
from bfcl.model_handler.api_inference.mining import MiningHandler
5+
6+
class DMCitoHandler(MiningHandler):
7+
def __init__(self, model_name, temperature) -> None:
8+
super().__init__(model_name, temperature)
9+
self.model_style = ModelStyle.OpenAI
10+
self.client = OpenAI(
11+
base_url= os.getenv("DMCITO_BASE_URL"),
12+
api_key=os.getenv("DMCITO_API_KEY"),
13+
)

berkeley-function-call-leaderboard/bfcl/model_handler/api_inference/mining.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def __init__(self, model_name, temperature) -> None:
1616
super().__init__(model_name, temperature)
1717
self.model_style = ModelStyle.OpenAI
1818
self.client = OpenAI(
19-
base_url="http://algo-dm-generation.mlamp.cn:8188/v1",
19+
base_url= os.getenv("MINING_BASE_URL"),
2020
api_key=os.getenv("MINING_API_KEY"),
2121
)
2222

0 commit comments

Comments
 (0)