| 123456789101112131415161718192021222324252627 |
- from langchain_community.embeddings import DashScopeEmbeddings
- from langchain_community.embeddings import OpenAIEmbeddings
- import os
- def get_embedding(embedding: str):
- # if embedding == "m3e":
- # embedding = OpenAIEmbeddings(
- # model="m3e",
- # api_key="sk-ZcS0IOaaR5FxEXAO2aEb9a37A7Cd41Fd9a70C74a854c20Af",
- # base_url="http://192.168.3.233:3001/v1",
- # headers={'content-type': 'application/json'}
- # )
- # try:
- # response = embedding.client.create(
- # model="m3e",
- # input=["Sample text for embedding"]
- # )
- # print("Response:", response)
- return DashScopeEmbeddings(dashscope_api_key="sk-86d4622141d74e9a8d7c38ee873c4d91")
- # except Exception as e:
- # print(f"Error: {e}")
- # raise
- # else:
- # raise ValueError(f"embedding {embedding} not support ")
|
