mirror of
https://github.com/mindverse/Second-Me.git
synced 2026-07-22 23:53:27 +00:00
2483288348
* Optimization of files Local Chat API.md and Public Chat API.md * MCP sever * Embedding strategy * change name * drop mcp files * llm.py change
14 lines
685 B
Python
14 lines
685 B
Python
from lpm_kernel.api.dto.user_llm_config_dto import (
|
|
UserLLMConfigDTO,
|
|
)
|
|
from typing import Optional
|
|
import lpm_kernel.common.strategy.strategy_openai as openai
|
|
import lpm_kernel.common.strategy.strategy_huggingface as huggingface
|
|
|
|
def strategy_classification(user_llm_config: Optional[UserLLMConfigDTO], chunked_texts):
|
|
if "sentence-transformers" in user_llm_config.embedding_endpoint:
|
|
# Using Hugging Face strategy to generate embedding vectors
|
|
return huggingface.huggingface_strategy(user_llm_config, chunked_texts)
|
|
else:
|
|
# Using openai strategy to generate embedding vectors
|
|
return openai.openai_strategy(user_llm_config, chunked_texts)
|