lostinthesea commited on
Commit
29ddc04
·
verified ·
1 Parent(s): 75e4bc2

Update agent.py

Browse files

change max tokens

Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -112,7 +112,7 @@ def process_pdf(state: AnkiGeneratorState) -> AnkiGeneratorState:
112
  pdf_b64 = _file_to_b64(state["pdf_file"])
113
  message = client.messages.create(
114
  model=ANTHROPIC_MODEL,
115
- max_tokens=2048,
116
  messages=[
117
  {
118
  "role": "user",
@@ -139,7 +139,7 @@ def process_image(state: AnkiGeneratorState) -> AnkiGeneratorState:
139
  mime = mimetypes.guess_type(state["img_file"])[0] or "image/png"
140
  message = client.messages.create(
141
  model=ANTHROPIC_MODEL,
142
- max_tokens=2048,
143
  messages=[
144
  {
145
  "role": "user",
@@ -173,7 +173,7 @@ def process_url(state: AnkiGeneratorState) -> AnkiGeneratorState:
173
  text = raw.decode("utf-8", errors="ignore")[:15000]
174
  message = client.messages.create(
175
  model=ANTHROPIC_MODEL,
176
- max_tokens=1024,
177
  messages=[
178
  {"role": "user", "content": [{"type": "text", "text": text}, {"type": "text", "text": _prompt("webpage", state)}]},
179
  ],
 
112
  pdf_b64 = _file_to_b64(state["pdf_file"])
113
  message = client.messages.create(
114
  model=ANTHROPIC_MODEL,
115
+ max_tokens=10240,
116
  messages=[
117
  {
118
  "role": "user",
 
139
  mime = mimetypes.guess_type(state["img_file"])[0] or "image/png"
140
  message = client.messages.create(
141
  model=ANTHROPIC_MODEL,
142
+ max_tokens=10240,
143
  messages=[
144
  {
145
  "role": "user",
 
173
  text = raw.decode("utf-8", errors="ignore")[:15000]
174
  message = client.messages.create(
175
  model=ANTHROPIC_MODEL,
176
+ max_tokens=10240,
177
  messages=[
178
  {"role": "user", "content": [{"type": "text", "text": text}, {"type": "text", "text": _prompt("webpage", state)}]},
179
  ],