Skip to content

Commit 23178cb

Browse files
committed
feat(ai): Allow to create df from chat response
jira: GDAI-238 risk: low
1 parent 01468bd commit 23178cb

File tree

4 files changed

+2576
-0
lines changed

4 files changed

+2576
-0
lines changed

gooddata-pandas/gooddata_pandas/dataframe.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,21 @@ def for_visualization(self, visualization_id: str, auto_index: bool = True) -> p
183183

184184
return self.for_items(columns, filter_by=filter_by, auto_index=auto_index)
185185

186+
def for_created_visualization(
187+
self, created_visualizations_response: dict
188+
) -> tuple[pandas.DataFrame, DataFrameMetadata]:
189+
"""
190+
Creates a data frame using a created visualization.
191+
192+
Args:
193+
created_visualizations_response (dict): Created visualization response.
194+
195+
Returns:
196+
pandas.DataFrame: A DataFrame instance.
197+
"""
198+
execution_definition = self._sdk.compute.build_exec_def_from_chat_result(created_visualizations_response)
199+
return self.for_exec_def(exec_def=execution_definition)
200+
186201
def result_cache_metadata_for_exec_result_id(self, result_id: str) -> ResultCacheMetadata:
187202
"""
188203
Retrieves result cache metadata for given :result_id:

0 commit comments

Comments
 (0)