MachineLearningReply/q-and-a-tool

1U

2�Te��@s$
dd
lZdd
l
Z
dd
l
Z
ddlmZ
e�
ejdd�
ZdZejdeddd	�
ejd3dd�
e
�d
d�e
�dd�e
�dd�e
�d�
e
�d�
d�Z	e
�dd�e
�dd�e
�dd�e
�dd�e
�dd�e
�d d!�e
�d"d!�e
�d#d$�e
�d%d&�e
�d'd(�e
�d)d�e
�d*d+�e
�d,d(�e
�d-d�d.�Z4d
S)/�N)
�load_dotenvzThis app lists animals)
�description)�inmemory�weaviateZmilvusZ5opensearchz--storerz.DocumentStore selection (default: %(default)s))�choices�default�helpz--namez
My Search App)
r�EMBEDDING_MODELz'sentence-transformers/all-MiniLM-L12-v2�GENERATIVE_MODELzgpt-4�EXTRACTIVE_MODELzdeepset/roberta-base-squad2�6OPENAI_KEY�7COHERE_KEY)r	r8rrr
�
WEAVIATE_HOSTzhttp://localhost�
WEAVIATE_PORTi��WEAVIATE_INDEXZDocument�WEAVIATE_EMBEDDING_DIMi�OPENSEARCH_SCHEME�https�OPENSEARCH_USERNAMEZadmin�OPENSEARCH_PASSWORD�OPENSEARCH_HOST�	localhost�OPENSEARCH_PORTi�#�OPENSEARCH_INDEX�document�OPENSEARCH_EMBEDDING_DIM�9MILVUS_URIzhttp://localhost:19530/default�MILVUS_INDEX�MILVUS_EMBEDDING_DIM)rrrrrrrrrrrrrr)�argparse�osZdotenvr�ArgumentParser�parserZdocument_store_choices�add_argument�getenvZ
model_configsZdocument_store_configs�r%r%�;/Users/s.kar/PycharmProjects/search_mlReply/utils/config.py�<module>
s:





10
11
12

�13 14
15
16
1718
19
20
21
22
23
2425
26
27�