config.py

import registry

from utils import get_widget_index_name

widget_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name=get_widget_index_name(),
    text_key="content",
)
app_info_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name="AppInfoV1",
    text_key="question",
    extra_keys=["answer", "suggested_follow_ups"],
)
scraped_sites_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name="IndexV1",
    text_key="content",
    extra_keys=["url"],
)

dapps_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name="Web3Apps",
    text_key="description",
    extra_keys=["url","name"],
)

api_docs_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name="APIDocsV1",
    text_key="description",
    extra_keys=["spec"],
)

crypto_tokens_index = dict(
    type="index.weaviate.WeaviateIndex",
    index_name="CryptoTokensV1",
    text_key="canonical_id",
    extra_keys=["name", "symbol"],
)


default_config = dict(
    type="system.System",
    chat=dict(
        type="chat.chatgpt_function_call.ChatGPTFunctionCallChat",
        model_name='gpt-4-0613',
        widget_index=widget_index,
        top_k=32,
    )
)


def initialize(cfg):
    if isinstance(cfg, dict):
        # recursively initialize objects
        _cfg = {k: initialize(v) for k, v in cfg.items()}
        if 'type' in _cfg and not _cfg.get('_streaming'):
            _type = _cfg.pop('type')
            _cls = registry.get_class(_type)
            print(f'Initializing instance of type: {_type}')
            return _cls(**_cfg)
        return _cfg
    elif isinstance(cfg, list):
        # recursively initialize items in list
        _cfg = [initialize(v) for v in cfg]
        return _cfg

    return cfg


def initialize_streaming(cfg, new_token_handler):
    if isinstance(cfg, dict):
        # recursively initialize objects
        _cfg = {k: initialize_streaming(v, new_token_handler) for k, v in cfg.items()}
        if 'type' in _cfg and _cfg.get('_streaming'):
            _type = _cfg.pop('type')
            _cls = registry.get_class(_type)
            print(f'Initializing instance of type: {_type}')
            return _cls(new_token_handler=new_token_handler, **_cfg)
        return _cfg
    elif isinstance(cfg, list):
        # recursively initialize items in list
        _cfg = [initialize_streaming(v, new_token_handler) for v in cfg]
        return _cfg

    return cfg


def initialize_system(config):
    return initialize(config)