Skip to content

Commit f1577f1

Browse files
ajroetkerajroetker
committed
Update autogen
1 parent cc49722 commit f1577f1

3 files changed

Lines changed: 33 additions & 33 deletions

File tree

‎src/antfly/client_generated/models/__init__.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@
107107
from .followup_step_config import FollowupStepConfig
108108
from .fuzziness_type_1 import FuzzinessType1
109109
from .fuzzy_query import FuzzyQuery
110+
from .generate_result import GenerateResult
110111
from .generator_config import GeneratorConfig
111112
from .generator_provider import GeneratorProvider
112113
from .geo_bounding_box_query import GeoBoundingBoxQuery
@@ -210,7 +211,6 @@
210211
from .shard_config import ShardConfig
211212
from .storage_status import StorageStatus
212213
from .success_message import SuccessMessage
213-
from .summarize_result import SummarizeResult
214214
from .sync_level import SyncLevel
215215
from .table import Table
216216
from .table_backup_status import TableBackupStatus
@@ -354,6 +354,7 @@
354354
"FollowupStepConfig",
355355
"FuzzinessType1",
356356
"FuzzyQuery",
357+
"GenerateResult",
357358
"GeneratorConfig",
358359
"GeneratorProvider",
359360
"GeoBoundingBoxQuery",
@@ -457,7 +458,6 @@
457458
"ShardConfig",
458459
"StorageStatus",
459460
"SuccessMessage",
460-
"SummarizeResult",
461461
"SyncLevel",
462462
"Table",
463463
"TableBackupStatus",

src/antfly/client_generated/models/summarize_result.py renamed to src/antfly/client_generated/models/generate_result.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -4,30 +4,30 @@
44
from attrs import define as _attrs_define
55
from attrs import field as _attrs_field
66

7-
T = TypeVar("T", bound="SummarizeResult")
7+
T = TypeVar("T", bound="GenerateResult")
88

99

1010
@_attrs_define
11-
class SummarizeResult:
12-
"""Result of a summarization operation. The summary is formatted as markdown with inline resource references using
13-
[resource_id <id>] or [resource_id <id1>, <id2>] format.
11+
class GenerateResult:
12+
"""Result of a generate operation. Formatted as markdown by default with inline resource references using [resource_id
13+
<id>] or [resource_id <id1>, <id2>] format.
1414
1515
Attributes:
16-
summary (str): The generated summary text in markdown format with inline resource references like [resource_id
17-
res1] or [resource_id res1, res2]
16+
text (str): The generated text in markdown format with inline resource references like [resource_id res1] or
17+
[resource_id res1, res2]
1818
"""
1919

20-
summary: str
20+
text: str
2121
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
2222

2323
def to_dict(self) -> dict[str, Any]:
24-
summary = self.summary
24+
text = self.text
2525

2626
field_dict: dict[str, Any] = {}
2727
field_dict.update(self.additional_properties)
2828
field_dict.update(
2929
{
30-
"summary": summary,
30+
"text": text,
3131
}
3232
)
3333

@@ -36,14 +36,14 @@ def to_dict(self) -> dict[str, Any]:
3636
@classmethod
3737
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
3838
d = dict(src_dict)
39-
summary = d.pop("summary")
39+
text = d.pop("text")
4040

41-
summarize_result = cls(
42-
summary=summary,
41+
generate_result = cls(
42+
text=text,
4343
)
4444

45-
summarize_result.additional_properties = d
46-
return summarize_result
45+
generate_result.additional_properties = d
46+
return generate_result
4747

4848
@property
4949
def additional_keys(self) -> list[str]:

‎src/antfly/client_generated/models/rag_result.py‎

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -8,27 +8,27 @@
88

99
if TYPE_CHECKING:
1010
from ..models.eval_result import EvalResult
11+
from ..models.generate_result import GenerateResult
1112
from ..models.query_result import QueryResult
12-
from ..models.summarize_result import SummarizeResult
1313

1414

1515
T = TypeVar("T", bound="RAGResult")
1616

1717

1818
@_attrs_define
1919
class RAGResult:
20-
"""RAG result with individual query results and summary
20+
"""RAG result with individual query results and generation/evaluation outcome
2121
2222
Attributes:
2323
query_results (Union[Unset, list['QueryResult']]): Results from each query. Check each result's status and error
2424
fields for failures.
25-
summary_result (Union[Unset, SummarizeResult]): Result of a summarization operation. The summary is formatted as
26-
markdown with inline resource references using [resource_id <id>] or [resource_id <id1>, <id2>] format.
25+
generate_result (Union[Unset, GenerateResult]): Result of a generate operation. Formatted as markdown by default
26+
with inline resource references using [resource_id <id>] or [resource_id <id1>, <id2>] format.
2727
eval_result (Union[Unset, EvalResult]): Complete evaluation result
2828
"""
2929

3030
query_results: Union[Unset, list["QueryResult"]] = UNSET
31-
summary_result: Union[Unset, "SummarizeResult"] = UNSET
31+
generate_result: Union[Unset, "GenerateResult"] = UNSET
3232
eval_result: Union[Unset, "EvalResult"] = UNSET
3333
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
3434

@@ -40,9 +40,9 @@ def to_dict(self) -> dict[str, Any]:
4040
query_results_item = query_results_item_data.to_dict()
4141
query_results.append(query_results_item)
4242

43-
summary_result: Union[Unset, dict[str, Any]] = UNSET
44-
if not isinstance(self.summary_result, Unset):
45-
summary_result = self.summary_result.to_dict()
43+
generate_result: Union[Unset, dict[str, Any]] = UNSET
44+
if not isinstance(self.generate_result, Unset):
45+
generate_result = self.generate_result.to_dict()
4646

4747
eval_result: Union[Unset, dict[str, Any]] = UNSET
4848
if not isinstance(self.eval_result, Unset):
@@ -53,8 +53,8 @@ def to_dict(self) -> dict[str, Any]:
5353
field_dict.update({})
5454
if query_results is not UNSET:
5555
field_dict["query_results"] = query_results
56-
if summary_result is not UNSET:
57-
field_dict["summary_result"] = summary_result
56+
if generate_result is not UNSET:
57+
field_dict["generate_result"] = generate_result
5858
if eval_result is not UNSET:
5959
field_dict["eval_result"] = eval_result
6060

@@ -63,8 +63,8 @@ def to_dict(self) -> dict[str, Any]:
6363
@classmethod
6464
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
6565
from ..models.eval_result import EvalResult
66+
from ..models.generate_result import GenerateResult
6667
from ..models.query_result import QueryResult
67-
from ..models.summarize_result import SummarizeResult
6868

6969
d = dict(src_dict)
7070
query_results = []
@@ -74,12 +74,12 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
7474

7575
query_results.append(query_results_item)
7676

77-
_summary_result = d.pop("summary_result", UNSET)
78-
summary_result: Union[Unset, SummarizeResult]
79-
if isinstance(_summary_result, Unset):
80-
summary_result = UNSET
77+
_generate_result = d.pop("generate_result", UNSET)
78+
generate_result: Union[Unset, GenerateResult]
79+
if isinstance(_generate_result, Unset):
80+
generate_result = UNSET
8181
else:
82-
summary_result = SummarizeResult.from_dict(_summary_result)
82+
generate_result = GenerateResult.from_dict(_generate_result)
8383

8484
_eval_result = d.pop("eval_result", UNSET)
8585
eval_result: Union[Unset, EvalResult]
@@ -90,7 +90,7 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
9090

9191
rag_result = cls(
9292
query_results=query_results,
93-
summary_result=summary_result,
93+
generate_result=generate_result,
9494
eval_result=eval_result,
9595
)
9696

0 commit comments

Comments
 (0)