Skip to content

Commit e1b30fc

Browse files
add external
1 parent 168ce66 commit e1b30fc

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

src/together/resources/evaluation.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def create(
3232
judge_system_template: str,
3333
input_data_file_path: str,
3434
judge_external_api_token: Optional[str] = None,
35+
judge_external_base_url: Optional[List[str]] = None,
3536
# Classify-specific parameters
3637
labels: Optional[List[str]] = None,
3738
pass_labels: Optional[List[str]] = None,
@@ -55,6 +56,7 @@ def create(
5556
judge_system_template: System template for the judge
5657
input_data_file_path: Path to input data file
5758
judge_external_api_token: Optional external API token for the judge model
59+
judge_external_base_url: Optional external base URLs for the judge model
5860
labels: List of classification labels (required for classify)
5961
pass_labels: List of labels considered as passing (required for classify)
6062
min_score: Minimum score value (required for score)
@@ -82,6 +84,7 @@ def create(
8284
model_source=judge_model_source,
8385
system_template=judge_system_template,
8486
external_api_token=judge_external_api_token,
87+
external_base_url=judge_external_base_url,
8588
)
8689
parameters: Union[ClassifyParameters, ScoreParameters, CompareParameters]
8790
# Build parameters based on type
@@ -423,6 +426,7 @@ async def create(
423426
judge_system_template: str,
424427
input_data_file_path: str,
425428
judge_external_api_token: Optional[str] = None,
429+
judge_external_base_url: Optional[List[str]] = None,
426430
# Classify-specific parameters
427431
labels: Optional[List[str]] = None,
428432
pass_labels: Optional[List[str]] = None,
@@ -446,6 +450,7 @@ async def create(
446450
judge_system_template: System template for the judge
447451
input_data_file_path: Path to input data file
448452
judge_external_api_token: Optional external API token for the judge model
453+
judge_external_base_url: Optional external base URLs for the judge model
449454
labels: List of classification labels (required for classify)
450455
pass_labels: List of labels considered as passing (required for classify)
451456
min_score: Minimum score value (required for score)
@@ -473,6 +478,7 @@ async def create(
473478
model_source=judge_model_source,
474479
system_template=judge_system_template,
475480
external_api_token=judge_external_api_token,
481+
external_base_url=judge_external_base_url,
476482
)
477483
parameters: Union[ClassifyParameters, ScoreParameters, CompareParameters]
478484
# Build parameters based on type

src/together/types/evaluation.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class JudgeModelConfig(BaseModel):
2727
model_source: Literal["serverless", "dedicated", "external"]
2828
system_template: str
2929
external_api_token: Optional[str] = None
30+
external_base_url: Optional[List[str]] = None
3031

3132

3233
class ModelRequest(BaseModel):
@@ -37,6 +38,7 @@ class ModelRequest(BaseModel):
3738
system_template: str
3839
input_template: str
3940
external_api_token: Optional[str] = None
41+
external_base_url: Optional[List[str]] = None
4042

4143

4244
class ClassifyParameters(BaseModel):

0 commit comments

Comments
 (0)