MissSqui commited on
Commit
dda0320
·
verified ·
1 Parent(s): 0be4a76

Update abc

Browse files
Files changed (1) hide show
  1. abc +23 -1
abc CHANGED
@@ -402,4 +402,26 @@ if __name__ == "__main__":
402
  process_pdf(r"C:\Users\shalini\Desktop\Project\abc.pdf")
403
 
404
 
405
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
402
  process_pdf(r"C:\Users\shalini\Desktop\Project\abc.pdf")
403
 
404
 
405
+ =======================================
406
+ def openai_llm_call(prompt: str, persona: str) -> str:
407
+
408
+ client = AzureOpenAI(
409
+ api_version=os.getenv("API_VERSION"),
410
+ azure_endpoint=os.getenv("ENDPOINT"),
411
+ api_key=os.getenv("OPENAI_API_KEY"),
412
+ )
413
+ response = client.chat.completions.create(
414
+ messages=[
415
+ {
416
+ "role": "system",
417
+ "content": persona,
418
+ },
419
+ {
420
+ "role": "user",
421
+ "content": prompt,
422
+ }
423
+ ],
424
+ model=os.getenv("DEPLOYMENT")
425
+ )
426
+
427
+ return response.choices[0].message.content