OwusuBlessing
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import anthropic
|
|
import os
|
|
from dotenv import load_dotenv
|
|
import json
|
|
from src.prompt import general_summary_prompt,custom_template_prompt
|
|
load_dotenv()
|
|
|
|
def general_summary(transcription):
|
|
client = anthropic.Anthropic(
|
|
api_key=os.getenv("ANTHTROPIC_API_KEY"),
|
|
)
|
|
message = client.messages.create(
|
|
model="claude-3-5-sonnet-20241022",
|
|
max_tokens=4000,
|
|
messages=[
|
|
{"role": "user", "content": f"{general_summary_prompt}"},
|
|
{"role": "user", "content": f"Transcription: {transcription}"}
|
|
]
|
|
)
|
|
|
|
text = message.content[0].text
|
|
return json.loads(text)
|
|
|
|
|
|
|
|
|
|
def custom_summary(template, transcription):
|
|
client = anthropic.Anthropic(
|
|
api_key=os.getenv("ANTHTROPIC_API_KEY"),
|
|
)
|
|
message = client.messages.create(
|
|
model="claude-3-5-sonnet-20241022",
|
|
max_tokens=8000,
|
|
messages=[
|
|
{"role": "user", "content": f"{custom_template_prompt}"},
|
|
{"role": "user", "content": f"TEMPLATE : {template}"},
|
|
{"role": "user", "content": f"Transcription: {transcription}"}
|
|
]
|
|
)
|
|
|
|
text = message.content[0].text
|
|
return json.loads(text)
|