[dev] del global key in config & modify workflow

This commit is contained in:
jinli.yl 2024-07-02 19:47:15 +08:00
parent c0d9a22574
commit 34d06868e2
3 changed files with 21 additions and 28 deletions

View file

@ -1,8 +1,6 @@
global_config:
language: cn
max_workers: 5
dash_scope_apikey:
open_ai_apikey:
memory_chat:
cli_memory_chat:
class: chat.cli_memory_chat
@ -23,7 +21,7 @@ memory_service:
description: "read session messages of the user"
read_memory:
class: memory.operation.read_memory
workflow: dummy_worker
workflow: set_query_worker,[extract_time_worker|retrieve_store_worker,semantic_rank_worker],fuse_rerank_worker
description: "read related memories of the user"
list_memory:
class: memory.operation.read_memory
@ -39,26 +37,6 @@ memory_service:
workflow: dummy_worker
description: "summary observation memories of the user"
interval_time: 300
models:
dashscope_generation:
class: models.llama_index_generation_model
module_name: dashscope_generation
model_name: qwen-max
dashscope_embedding:
class: models.llama_index_embedding_model
module_name: dashscope_embedding
model_name: text-embedding-v2
dashscope_rank:
class: models.llama_index_rank_model
module_name: dashscope_rank
model_name: gte-rerank
vector_store:
class: storage.llama_index_elastic_search_store
embedding_model: dashscope_embedding
index_name: memory_index
es_url: http://localhost:9200
monitor:
class: storage.dummy_monitor
worker:
dummy_worker:
class: memory.worker.dummy_worker
@ -95,3 +73,23 @@ worker:
generation_model_top_k: 1
today_obs_top_k: 30
contra_repeat_max_count: 50
models:
dashscope_generation:
class: models.llama_index_generation_model
module_name: dashscope_generation
model_name: qwen-max
dashscope_embedding:
class: models.llama_index_embedding_model
module_name: dashscope_embedding
model_name: text-embedding-v2
dashscope_rank:
class: models.llama_index_rank_model
module_name: dashscope_rank
model_name: gte-rerank
vector_store:
class: storage.llama_index_elastic_search_store
embedding_model: dashscope_embedding
index_name: memory_index
es_url: http://localhost:9200
monitor:
class: storage.dummy_monitor

View file

@ -78,5 +78,3 @@ class LlamaIndexGenerationModel(BaseModel):
response = await self.model.achat(**self.data)
results.raw = response
return results
raise NotImplementedError

View file

@ -3,15 +3,12 @@ import random
import re
import time
from copy import deepcopy
from datetime import datetime
from importlib import import_module
import pyfiglet
from termcolor import colored, COLORS
from memory_scope.constants.language_constants import WEEKDAYS
from memory_scope.enumeration.message_role_enum import MessageRoleEnum
from memory_scope.utils.global_context import G_CONTEXT
def underscore_to_camelcase(name: str, is_first_title: bool = True):