curategpt.adhoc package

Submodules

curategpt.adhoc.gocam_predictor module

GO-CAM predictor class.

class curategpt.adhoc.gocam_predictor.GOCAMPredictor(store=None, database_type='chromadb', database_path='gocam', collection_name='gocam', model_name='gpt-4o', extractor=None, include_standard_annotations=False, gocam_wrapper=<factory>, strict=False, **_kwargs)

Bases: object

collection_name: str = 'gocam'

database_path: str = 'gocam'

database_type: str = 'chromadb'

extractor: BasicExtractor = None

fix_yaml(yaml_text, e)

Fix the YAML text.

Parameters:

yaml_text (str) – The YAML text

Return type:

str

Returns:

The fixed YAML text

gocam_by_id(id)

Return type:

Dict[str, Any]

gocam_wrapper: GOCAMWrapper

include_standard_annotations: bool = False

model_name: str = 'gpt-4o'

predict_activity_unit(gocam, stub, num_examples=0)

Predict the missing activity in a GO-CAM model.

Parameters:

• gocam (Dict[str, Any]) – The GO-CAM model

• stub (Dict[str, Any]) – The stub to fill in

• num_examples – The number of examples to generate

Return type:

Dict[str, Any]

Returns:

The predicted activity

store: DBAdapter = None

strict: bool = False

Module contents