Skip to content

Add support for bidirectional streaming within eval service.#370

Open
erinlimbogoogle wants to merge 3 commits intomainfrom
erinbranch
Open

Add support for bidirectional streaming within eval service.#370
erinlimbogoogle wants to merge 3 commits intomainfrom
erinbranch

Conversation

@erinlimbogoogle
Copy link
Copy Markdown
Collaborator

No description provided.

Comment thread evalbench/dataset/cortadoinput.py Fixed
Comment thread evalbench/evaluator/cortadoorchestrator.py Fixed

message EvalInputRequest {
int64 id = 1;
string conversation_id = 18;
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

should be sorted by id

string sql_generator_error = 12;
float sql_generator_time = 13;
string generated_sql = 14;
string generated_nl_response = 19;
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

same

Comment thread evalbench/eval_service.py

finally:
# Clean up the global registry to prevent memory leaks.
PROXY_QUEUES.pop(session_id, None)
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

If a client connection drops forcefully or hangs, and the Interact RPC does not hit its finally block cleanup (or if the thread hangs indefinitely
blocked on inbound_response = thread_inbox.get(block=True, timeout=300.0) ), then that entry in PROXY_QUEUES will leak and remain in memory even
after the background reaper deletes the session from Disk/SessionManager!

we should clear it when the session gets deleted.

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