Update agent.py
Browse fileschange max tokens
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=
|
| 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=
|
| 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=
|
| 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 |
],
|