diff options
Diffstat (limited to 'src/server/chunker/pdf_chunker.py')
| -rw-r--r-- | src/server/chunker/pdf_chunker.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/server/chunker/pdf_chunker.py b/src/server/chunker/pdf_chunker.py index 9d4cfb1b2..feb437f1f 100644 --- a/src/server/chunker/pdf_chunker.py +++ b/src/server/chunker/pdf_chunker.py @@ -602,12 +602,11 @@ class PDFChunker: try: response = self.client.messages.create( - model='claude-3-5-sonnet-20240620', + model='claude-3-7-sonnet-20250219', system=prompt, max_tokens=400 * len(images), # Increased token limit for more detailed summaries messages=messages, temperature=0, - extra_headers={"anthropic-beta": "max-tokens-3-5-sonnet-2024-07-15"} ) # Parse the response @@ -767,7 +766,7 @@ class Document: client = OpenAI() # Initialize OpenAI client for text generation completion = client.chat.completions.create( - model="gpt-3.5-turbo", # Specify the language model + model="gpt-4o", # Specify the language model messages=[ {"role": "system", "content": "You are an AI assistant tasked with summarizing a document. You are provided with important chunks from the document and provide a summary, as best you can, of what the document will contain overall. Be concise and brief with your response."}, |