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

Improve LM configuration serialization and deserialization for CoSTORM #329

Open
wants to merge 7 commits into
base: main
Choose a base branch
from

Conversation

JamesHWade
Copy link

This PR enhances the serialization and deserialization of language model configurations in the CoStormRunner class. It improves persistence by properly saving and restoring LM configurations and introduces proper handling of authentication credentials.

Changes

  • Implemented improved to_dict() and from_dict() methods in CollaborativeStormLMConfigs
  • Added security measures to avoid serializing sensitive authentication data
  • Improved CoStormRunner serialization to properly handle LM configurations
  • Added support for persisting the report generated by knowledge base
  • Ensured Azure authentication credentials are properly handled during deserialization

Benefits

  • More reliable serialization/deserialization of session state
  • Better security by not persisting sensitive authentication information
  • Improved session restoration with proper LM configuration persistence
  • Support for saving and loading completed reports

Testing

The changes have been tested by serializing and deserializing CoStormRunner instances with various configurations and verifying that the reconstructed instances maintain their functionality.

Fixes #328

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.

[BUG] LM configurations lost when reloading a saved CoSTORM runner
1 participant