Skip to content

Register connection if connection_config is included ModuleConfig#362

Draft
mynhardtburger wants to merge 1 commit into
caikit:mainfrom
mynhardtburger:connection_config-init-args
Draft

Register connection if connection_config is included ModuleConfig#362
mynhardtburger wants to merge 1 commit into
caikit:mainfrom
mynhardtburger:connection_config-init-args

Conversation

@mynhardtburger
Copy link
Copy Markdown
Contributor

No description provided.

Signed-off-by: Mynhardt Burger <Mynhardt.Burger@ibm.com>
"prompt_cache_id": self._prompt_cache_id,
"eos_token": self.eos_token,
"verbalizer": self.verbalizer,
# TODO: Add connection_config
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Should the save() method not also save the existing connection_config for future use?

"sep_token": self._sep_token,
"eos_token": self._eos_token,
"pad_token": self._pad_token,
# TODO: Add connection_config
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Should the save() method not also save the existing connection_config for future use?

eos_token=eos_token,
pad_token=pad_token,
tgis_backend=load_backend,
# TODO: Add connection_config?
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

When bootstrapping should the connection_config also be populated?

log.debug("TGIS model ID: %s", base_model_name)

# connection_config is a dict with all the necessary fields to successfully create
# a TGISConnection using the .from_config() method.
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

The connection config will roughly look like this:

connection_config:
  hostname: 
  model_id: 
  namespace: 
  ca_cert_file: 
  client_cert_file: 
  client_key_file: 
  prompt_dir: 
  grpc_lb_policy_name: 
  grpc_lb_poll_interval_s: 
  hostname_override: 

@mynhardtburger
Copy link
Copy Markdown
Contributor Author

@gabe-l-hart FYI

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.

1 participant