Compare commits

...

1 Commits

Author SHA1 Message Date
Toran Bruce Richards
d681083880 fix exa contents advanced toggle 2025-05-17 20:12:01 +01:00

View File

@@ -1,6 +1,6 @@
from typing import List from typing import List
from pydantic import BaseModel
from backend.blocks.exa._auth import ( from backend.blocks.exa._auth import (
ExaCredentials, ExaCredentials,
@@ -9,29 +9,14 @@ from backend.blocks.exa._auth import (
) )
from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema
from backend.data.model import SchemaField from backend.data.model import SchemaField
from .helpers import ContentSettings
from backend.util.request import requests from backend.util.request import requests
class ContentRetrievalSettings(BaseModel): # Use the fully-typed ``ContentSettings`` model from ``helpers`` instead of
text: dict = SchemaField( # anonymous ``dict`` fields so that the generated JSON schema contains proper
description="Text content settings", # object properties. Without this, the frontend crashes when rendering the
default={"maxCharacters": 1000, "includeHtmlTags": False}, # advanced settings for this block.
advanced=True,
)
highlights: dict = SchemaField(
description="Highlight settings",
default={
"numSentences": 3,
"highlightsPerUrl": 3,
"query": "",
},
advanced=True,
)
summary: dict = SchemaField(
description="Summary settings",
default={"query": ""},
advanced=True,
)
class ExaContentsBlock(Block): class ExaContentsBlock(Block):
@@ -40,9 +25,9 @@ class ExaContentsBlock(Block):
ids: List[str] = SchemaField( ids: List[str] = SchemaField(
description="Array of document IDs obtained from searches", description="Array of document IDs obtained from searches",
) )
contents: ContentRetrievalSettings = SchemaField( contents: ContentSettings = SchemaField(
description="Content retrieval settings", description="Content retrieval settings",
default=ContentRetrievalSettings(), default=ContentSettings(),
advanced=True, advanced=True,
) )