ascender1729/pavan/fix-pydantic-version-compatibility-issues

Fix field_validator syntax for Pydantic v2
pull/878/head
Pavan Kumar 3 weeks ago committed by GitHub
commit 2ae4b5f43b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -49,7 +49,7 @@ class RAGConfig(BaseModel):
default=None, description="Keywords to check for relevance" default=None, description="Keywords to check for relevance"
) )
@field_validator("relevance_keywords", pre=True) @field_validator("relevance_keywords", mode="before")
def set_default_keywords(cls, v): def set_default_keywords(cls, v):
if v is None: if v is None:
return [ return [
@ -227,9 +227,12 @@ class AgentRAGHandler:
formatted_sections = [header] formatted_sections = [header]
for i, result in enumerate(results, 1): for i, result in enumerate(results, 1):
content, score, source, metadata = ( (
self._extract_result_fields(result) content,
) score,
source,
metadata,
) = self._extract_result_fields(result)
section = f""" section = f"""
[Memory {i}] Relevance: {score} | Source: {source} [Memory {i}] Relevance: {score} | Source: {source}

Loading…
Cancel
Save