gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update lightrag_azure_openai_demo.py

Browse Source 
更新代码格式
This commit is contained in:
Magic_yuan
2024-11-21 18:44:38 +08:00
committed by GitHub
parent c0b040fe1b
commit 2e8768b301
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/lightrag_azure_openai_demo.py
View File

@@ -31,13 +31,12 @@ os.mkdir(WORKING_DIR)
async def llm_model_func( async def llm_model_func(
prompt, system_prompt=None, history_messages=[], **kwargs prompt, system_prompt=None, history_messages=[], **kwargs
) -> str: ) -> str:
client = AzureOpenAI( client = AzureOpenAI(
api_key=AZURE_OPENAI_API_KEY, api_key=LLM_AZURE_OPENAI_KEY,
api_version=AZURE_OPENAI_API_VERSION, api_version=LLM_AZURE_OPENAI_VERSION,
azure_endpoint=AZURE_OPENAI_ENDPOINT azure_endpoint=LLM_AZURE_OPENAI_API
) )
messages = [] messages = []
@@ -48,7 +47,7 @@ async def llm_model_func(
messages.append({"role": "user", "content": prompt}) messages.append({"role": "user", "content": prompt})
chat_completion = client.chat.completions.create( chat_completion = client.chat.completions.create(
model=AZURE_OPENAI_DEPLOYMENT, # model = "deployment_name". model=LLM_AZURE_OPENAI_DEPLOYMENT, # model = "deployment_name".
messages=messages, messages=messages,
temperature=kwargs.get("temperature", 0), temperature=kwargs.get("temperature", 0),
top_p=kwargs.get("top_p", 1), top_p=kwargs.get("top_p", 1),
@@ -58,7 +57,6 @@ async def llm_model_func(
async def embedding_func(texts: list[str]) -> np.ndarray: async def embedding_func(texts: list[str]) -> np.ndarray:
client = AzureOpenAI( client = AzureOpenAI(
api_key=AZURE_OPENAI_API_KEY, api_key=AZURE_OPENAI_API_KEY,
api_version=AZURE_EMBEDDING_API_VERSION, api_version=AZURE_EMBEDDING_API_VERSION,
@@ -68,7 +66,7 @@ async def embedding_func(texts: list[str]) -> np.ndarray:
model=AZURE_EMBEDDING_DEPLOYMENT, model=AZURE_EMBEDDING_DEPLOYMENT,
input=texts input=texts
) )
embeddings = [item.embedding for item in embedding.data] embeddings = [item.embedding for item in embedding.data]
return np.array(embeddings) return np.array(embeddings)