Skip to content

refactor: rename gepa into prompt optimization #1034

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
leonardmq merged 2 commits into from
Feb 16, 2026
Merged

refactor: rename gepa into prompt optimization #1034

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a3bd11e
refactor: rename gepa into prompt optimization
leonardmq Feb 16, 2026
0ef7d19
fix: return exceptions
leonardmq Feb 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions app/desktop/desktop_server.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,9 @@
from app.desktop.studio_server.dev_tools import connect_dev_tools
from app.desktop.studio_server.eval_api import connect_evals_api
from app.desktop.studio_server.finetune_api import connect_fine_tune_api
from app.desktop.studio_server.gepa_job_api import connect_gepa_job_api
from app.desktop.studio_server.prompt_optimization_job_api import (
connect_prompt_optimization_job_api,
)
from app.desktop.studio_server.import_api import connect_import_api
from app.desktop.studio_server.prompt_api import connect_prompt_api
from app.desktop.studio_server.provider_api import connect_provider_api
Expand Down Expand Up @@ -66,7 +68,7 @@ def make_app(tk_root: tk.Tk | None = None):
connect_evals_api(app)
connect_import_api(app, tk_root=tk_root)
connect_tool_servers_api(app)
connect_gepa_job_api(app)
connect_prompt_optimization_job_api(app)
connect_copilot_api(app)
connect_dev_tools(app)

Expand Down
10 changes: 5 additions & 5 deletions ...obs_gepa_job_check_model_supported_get.py → ...mization_job_check_model_supported_get.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def _get_kwargs(

_kwargs: dict[str, Any] = {
"method": "get",
"url": "/v1/jobs/gepa_job/check_model_supported",
"url": "/v1/jobs/prompt_optimization_job/check_model_supported",
"params": params,
}

Expand Down Expand Up @@ -69,7 +69,7 @@ def sync_detailed(
model_name: str,
model_provider_name: str,
) -> Response[CheckModelSupportedResponse | HTTPValidationError]:
"""Check Model Supported
"""Check Prompt Optimization Model Supported
Args:
model_name (str):
Expand Down Expand Up @@ -101,7 +101,7 @@ def sync(
model_name: str,
model_provider_name: str,
) -> CheckModelSupportedResponse | HTTPValidationError | None:
"""Check Model Supported
"""Check Prompt Optimization Model Supported
Args:
model_name (str):
Expand All @@ -128,7 +128,7 @@ async def asyncio_detailed(
model_name: str,
model_provider_name: str,
) -> Response[CheckModelSupportedResponse | HTTPValidationError]:
"""Check Model Supported
"""Check Prompt Optimization Model Supported
Args:
model_name (str):
Expand Down Expand Up @@ -158,7 +158,7 @@ async def asyncio(
model_name: str,
model_provider_name: str,
) -> CheckModelSupportedResponse | HTTPValidationError | None:
"""Check Model Supported
"""Check Prompt Optimization Model Supported
Args:
model_name (str):
Expand Down
34 changes: 17 additions & 17 deletions ...ult_v1_jobs_gepa_job_job_id_result_get.py → ...mpt_optimization_job_job_id_result_get.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@

from ... import errors
from ...client import AuthenticatedClient, Client
from ...models.gepa_job_result_response import GEPAJobResultResponse
from ...models.http_validation_error import HTTPValidationError
from ...models.prompt_optimization_job_result_response import PromptOptimizationJobResultResponse
from ...types import Response


Expand All @@ -17,7 +17,7 @@ def _get_kwargs(

_kwargs: dict[str, Any] = {
"method": "get",
"url": "/v1/jobs/gepa_job/{job_id}/result".format(
"url": "/v1/jobs/prompt_optimization_job/{job_id}/result".format(
job_id=quote(str(job_id), safe=""),
),
}
Expand All @@ -27,9 +27,9 @@ def _get_kwargs(

def _parse_response(
*, client: AuthenticatedClient | Client, response: httpx.Response
) -> GEPAJobResultResponse | HTTPValidationError | None:
) -> HTTPValidationError | PromptOptimizationJobResultResponse | None:
if response.status_code == 200:
response_200 = GEPAJobResultResponse.from_dict(response.json())
response_200 = PromptOptimizationJobResultResponse.from_dict(response.json())

return response_200

Expand All @@ -46,7 +46,7 @@ def _parse_response(

def _build_response(
*, client: AuthenticatedClient | Client, response: httpx.Response
) -> Response[GEPAJobResultResponse | HTTPValidationError]:
) -> Response[HTTPValidationError | PromptOptimizationJobResultResponse]:
return Response(
status_code=HTTPStatus(response.status_code),
content=response.content,
Expand All @@ -59,8 +59,8 @@ def sync_detailed(
job_id: str,
*,
client: AuthenticatedClient,
) -> Response[GEPAJobResultResponse | HTTPValidationError]:
"""Get Gepa Job Result
) -> Response[HTTPValidationError | PromptOptimizationJobResultResponse]:
"""Get Prompt Optimization Job Result
Args:
job_id (str):
Expand All @@ -70,7 +70,7 @@ def sync_detailed(
httpx.TimeoutException: If the request takes longer than Client.timeout.
Returns:
Response[GEPAJobResultResponse | HTTPValidationError]
Response[HTTPValidationError | PromptOptimizationJobResultResponse]
"""

kwargs = _get_kwargs(
Expand All @@ -88,8 +88,8 @@ def sync(
job_id: str,
*,
client: AuthenticatedClient,
) -> GEPAJobResultResponse | HTTPValidationError | None:
"""Get Gepa Job Result
) -> HTTPValidationError | PromptOptimizationJobResultResponse | None:
"""Get Prompt Optimization Job Result
Args:
job_id (str):
Expand All @@ -99,7 +99,7 @@ def sync(
httpx.TimeoutException: If the request takes longer than Client.timeout.
Returns:
GEPAJobResultResponse | HTTPValidationError
HTTPValidationError | PromptOptimizationJobResultResponse
"""

return sync_detailed(
Expand All @@ -112,8 +112,8 @@ async def asyncio_detailed(
job_id: str,
*,
client: AuthenticatedClient,
) -> Response[GEPAJobResultResponse | HTTPValidationError]:
"""Get Gepa Job Result
) -> Response[HTTPValidationError | PromptOptimizationJobResultResponse]:
"""Get Prompt Optimization Job Result
Args:
job_id (str):
Expand All @@ -123,7 +123,7 @@ async def asyncio_detailed(
httpx.TimeoutException: If the request takes longer than Client.timeout.
Returns:
Response[GEPAJobResultResponse | HTTPValidationError]
Response[HTTPValidationError | PromptOptimizationJobResultResponse]
"""

kwargs = _get_kwargs(
Expand All @@ -139,8 +139,8 @@ async def asyncio(
job_id: str,
*,
client: AuthenticatedClient,
) -> GEPAJobResultResponse | HTTPValidationError | None:
"""Get Gepa Job Result
) -> HTTPValidationError | PromptOptimizationJobResultResponse | None:
"""Get Prompt Optimization Job Result
Args:
job_id (str):
Expand All @@ -150,7 +150,7 @@ async def asyncio(
httpx.TimeoutException: If the request takes longer than Client.timeout.
Returns:
GEPAJobResultResponse | HTTPValidationError
HTTPValidationError | PromptOptimizationJobResultResponse
"""

return (
Expand Down
32 changes: 17 additions & 15 deletions ...t_gepa_job_v1_jobs_gepa_job_start_post.py → ...obs_prompt_optimization_job_start_post.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,21 +5,23 @@

from ... import errors
from ...client import AuthenticatedClient, Client
from ...models.body_start_gepa_job_v1_jobs_gepa_job_start_post import BodyStartGepaJobV1JobsGepaJobStartPost
from ...models.body_start_prompt_optimization_job_v1_jobs_prompt_optimization_job_start_post import (
BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
)
from ...models.http_validation_error import HTTPValidationError
from ...models.job_start_response import JobStartResponse
from ...types import Response


def _get_kwargs(
*,
body: BodyStartGepaJobV1JobsGepaJobStartPost,
body: BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
) -> dict[str, Any]:
headers: dict[str, Any] = {}

_kwargs: dict[str, Any] = {
"method": "post",
"url": "/v1/jobs/gepa_job/start",
"url": "/v1/jobs/prompt_optimization_job/start",
}

_kwargs["files"] = body.to_multipart()
Expand Down Expand Up @@ -61,12 +63,12 @@ def _build_response(
def sync_detailed(
*,
client: AuthenticatedClient,
body: BodyStartGepaJobV1JobsGepaJobStartPost,
body: BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
) -> Response[HTTPValidationError | JobStartResponse]:
"""Start Gepa Job
"""Start Prompt Optimization Job
Args:
body (BodyStartGepaJobV1JobsGepaJobStartPost):
body (BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost):
Raises:
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
Expand All @@ -90,12 +92,12 @@ def sync_detailed(
def sync(
*,
client: AuthenticatedClient,
body: BodyStartGepaJobV1JobsGepaJobStartPost,
body: BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
) -> HTTPValidationError | JobStartResponse | None:
"""Start Gepa Job
"""Start Prompt Optimization Job
Args:
body (BodyStartGepaJobV1JobsGepaJobStartPost):
body (BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost):
Raises:
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
Expand All @@ -114,12 +116,12 @@ def sync(
async def asyncio_detailed(
*,
client: AuthenticatedClient,
body: BodyStartGepaJobV1JobsGepaJobStartPost,
body: BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
) -> Response[HTTPValidationError | JobStartResponse]:
"""Start Gepa Job
"""Start Prompt Optimization Job
Args:
body (BodyStartGepaJobV1JobsGepaJobStartPost):
body (BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost):
Raises:
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
Expand All @@ -141,12 +143,12 @@ async def asyncio_detailed(
async def asyncio(
*,
client: AuthenticatedClient,
body: BodyStartGepaJobV1JobsGepaJobStartPost,
body: BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
) -> HTTPValidationError | JobStartResponse | None:
"""Start Gepa Job
"""Start Prompt Optimization Job
Args:
body (BodyStartGepaJobV1JobsGepaJobStartPost):
body (BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost):
Raises:
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
Expand Down
14 changes: 8 additions & 6 deletions app/desktop/studio_server/api_client/kiln_ai_server_client/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,9 @@
from .answer_option import AnswerOption
from .answer_option_with_selection import AnswerOptionWithSelection
from .api_key_verification_result import ApiKeyVerificationResult
from .body_start_gepa_job_v1_jobs_gepa_job_start_post import BodyStartGepaJobV1JobsGepaJobStartPost
from .body_start_prompt_optimization_job_v1_jobs_prompt_optimization_job_start_post import (
BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost,
)
from .body_start_sample_job_v1_jobs_sample_job_start_post import BodyStartSampleJobV1JobsSampleJobStartPost
from .check_entitlements_v1_check_entitlements_get_response_check_entitlements_v1_check_entitlements_get import (
CheckEntitlementsV1CheckEntitlementsGetResponseCheckEntitlementsV1CheckEntitlementsGet,
Expand All @@ -16,8 +18,6 @@
from .generate_batch_input import GenerateBatchInput
from .generate_batch_output import GenerateBatchOutput
from .generate_batch_output_data_by_topic import GenerateBatchOutputDataByTopic
from .gepa_job_output import GEPAJobOutput
from .gepa_job_result_response import GEPAJobResultResponse
from .health_health_get_response_health_health_get import HealthHealthGetResponseHealthHealthGet
from .http_validation_error import HTTPValidationError
from .job_start_response import JobStartResponse
Expand All @@ -27,6 +27,8 @@
from .model_provider_name import ModelProviderName
from .new_proposed_spec_edit_api import NewProposedSpecEditApi
from .output_file_info import OutputFileInfo
from .prompt_optimization_job_output import PromptOptimizationJobOutput
from .prompt_optimization_job_result_response import PromptOptimizationJobResultResponse
from .question import Question
from .question_set import QuestionSet
from .question_with_answer import QuestionWithAnswer
Expand Down Expand Up @@ -55,7 +57,7 @@
"AnswerOption",
"AnswerOptionWithSelection",
"ApiKeyVerificationResult",
"BodyStartGepaJobV1JobsGepaJobStartPost",
"BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost",
"BodyStartSampleJobV1JobsSampleJobStartPost",
"CheckEntitlementsV1CheckEntitlementsGetResponseCheckEntitlementsV1CheckEntitlementsGet",
"CheckModelSupportedResponse",
Expand All @@ -66,8 +68,6 @@
"GenerateBatchInput",
"GenerateBatchOutput",
"GenerateBatchOutputDataByTopic",
"GEPAJobOutput",
"GEPAJobResultResponse",
"HealthHealthGetResponseHealthHealthGet",
"HTTPValidationError",
"JobStartResponse",
Expand All @@ -77,6 +77,8 @@
"ModelProviderName",
"NewProposedSpecEditApi",
"OutputFileInfo",
"PromptOptimizationJobOutput",
"PromptOptimizationJobResultResponse",
"Question",
"QuestionSet",
"QuestionWithAnswer",
Expand Down
10 changes: 5 additions & 5 deletions ...t_gepa_job_v1_jobs_gepa_job_start_post.py → ...obs_prompt_optimization_job_start_post.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,11 @@
from .. import types
from ..types import File

T = TypeVar("T", bound="BodyStartGepaJobV1JobsGepaJobStartPost")
T = TypeVar("T", bound="BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost")


@_attrs_define
class BodyStartGepaJobV1JobsGepaJobStartPost:
class BodyStartPromptOptimizationJobV1JobsPromptOptimizationJobStartPost:
"""
Attributes:
task_id (str): The task ID
Expand Down Expand Up @@ -79,15 +79,15 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:

project_zip = File(payload=BytesIO(d.pop("project_zip")))

body_start_gepa_job_v1_jobs_gepa_job_start_post = cls(
body_start_prompt_optimization_job_v1_jobs_prompt_optimization_job_start_post = cls(
task_id=task_id,
target_run_config_id=target_run_config_id,
eval_ids=eval_ids,
project_zip=project_zip,
)

body_start_gepa_job_v1_jobs_gepa_job_start_post.additional_properties = d
return body_start_gepa_job_v1_jobs_gepa_job_start_post
body_start_prompt_optimization_job_v1_jobs_prompt_optimization_job_start_post.additional_properties = d
return body_start_prompt_optimization_job_v1_jobs_prompt_optimization_job_start_post

@property
def additional_keys(self) -> list[str]:
Expand Down
12 changes: 6 additions & 6 deletions ...i_server_client/models/gepa_job_output.py → .../models/prompt_optimization_job_output.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,12 @@
from attrs import define as _attrs_define
from attrs import field as _attrs_field

T = TypeVar("T", bound="GEPAJobOutput")
T = TypeVar("T", bound="PromptOptimizationJobOutput")


@_attrs_define
class GEPAJobOutput:
"""Output from the GEPA job.
class PromptOptimizationJobOutput:
"""Output from the prompt optimization job.
Attributes:
optimized_prompt (str):
Expand All @@ -38,12 +38,12 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
d = dict(src_dict)
optimized_prompt = d.pop("optimized_prompt")

gepa_job_output = cls(
prompt_optimization_job_output = cls(
optimized_prompt=optimized_prompt,
)

gepa_job_output.additional_properties = d
return gepa_job_output
prompt_optimization_job_output.additional_properties = d
return prompt_optimization_job_output

@property
def additional_keys(self) -> list[str]:
Expand Down
Loading
Loading