Update abc
Browse files
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
|