Skip to content
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.

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

get full completion in cli completion download #272

Conversation

wenzhe-log10
Copy link
Collaborator

No description provided.

@@ -270,7 +270,15 @@ def download_completions(limit, offset, timeout, tags, from_date, to_date, file)
fetched_total += new_data_size

for completion in new_data:
output_file.write(json.dumps(completion) + "\n")
res = _get_completion(completion["id"])
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Completions()._get_completions() returns a list of completions with a compact info, such as last messages instead of full.
_get_completion(id) get the full completion by id with all messages.
So after we get the list of completions, we do a get for each of the ids for the full completion to save.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

vercel has limitation on how much data we can return, so we had to trim them. Otherwise, it would cause the request to failed 😞

@wenzhe-log10 wenzhe-log10 requested review from kxtran and nqn August 19, 2024 16:35
output_file.write(json.dumps(completion) + "\n")
res = _get_completion(completion["id"])
if res.status_code != 200:
rich.print(f"Error fetching completion {completion['id']}")
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

should we mention here to ask them to retry and then contact us for additional support?

Copy link
Collaborator

@kxtran kxtran left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🎉

@wenzhe-log10 wenzhe-log10 merged commit b8e6e1f into main Aug 19, 2024
2 checks passed
@wenzhe-log10 wenzhe-log10 deleted the wenzhe/eng-1044-debug-cli-completions-download-partial-messages branch August 19, 2024 16:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants