Skip to content

Commit 661f400

Browse files
Some fix
1 parent ea5891a commit 661f400

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

src/datacustomcode/llm_gateway/types/generate_text_request_builder.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,6 @@ def set_localization(self, localization: dict = None, locale: str = None):
5050
else:
5151
raise ValueError("Must provide either localization or locale")
5252

53-
self._localization = localization
5453
return self
5554

5655
def set_tags(self, tags: dict):

src/datacustomcode/llm_gateway/types/generate_text_response.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,13 @@ def is_error(self) -> bool:
3737
@property
3838
def text(self) -> str:
3939
"""Generated text (convenience property)."""
40-
if self.is_success:
40+
if self.is_success and self.data:
4141
return self.data.get('generation', {}).get('generatedText', '')
4242
return ''
4343

4444
@property
4545
def error_code(self) -> str:
4646
"""Generated text (convenience property)."""
47-
if self.is_error:
47+
if self.is_error and self.data:
4848
return self.data.get('errorCode', self.status_code)
4949
return ''

0 commit comments

Comments
 (0)