chore: Support for class-based `config` is deprecated, use ConfigDict instead

app/api/routers/chat.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 from typing import List, Any, Optional, Dict, Tuple
 from fastapi import APIRouter, Depends, HTTPException, Request, status
 from llama_index.core.chat_engine.types import BaseChatEngine
@@ -19,9 +19,8 @@ class _Message(BaseModel):
 
 class _ChatData(BaseModel):
     messages: List[_Message]
-
-    class Config:
-        json_schema_extra = {
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "messages": [
                     {
@@ -31,6 +30,7 @@ class _ChatData(BaseModel):
                 ]
             }
         }
+    )
 
 
 class _SourceNodes(BaseModel):
@@ -121,12 +121,7 @@ async def chat(
         yield VercelStreamResponse.convert_data(
             {
                 "type": "sources",
-                "data": {
-                    "nodes": [
-                        _SourceNodes.from_source_node(node).dict()
-                        for node in response.source_nodes
-                    ]
-                },
+                "data": {"nodes": [_SourceNodes.from_source_node(node).dict() for node in response.source_nodes]},
             }
         )
 

app/model/transaction.py

@@ -30,13 +30,7 @@ class Transaction(Base, BaseModel):
 
     @classmethod
     async def update(cls: "type[Transaction]", db: AsyncSession, id: int, **kwargs) -> "Transaction":
-        query = (
-            sql.update(cls)
-            .where(cls.id == id)
-            .values(**kwargs)
-            .execution_options(synchronize_session="fetch")
-            .returning(cls.id)
-        )
+        query = sql.update(cls).where(cls.id == id).values(**kwargs).execution_options(synchronize_session="fetch")
         transactions = await db.scalars(query)
         transaction = transactions.first()
         await db.commit()

app/schema/base.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 
 class PydanticBaseModel(BaseModel):
@@ -7,9 +7,8 @@ class PydanticBaseModel(BaseModel):
 
 
 class BaseModel(PydanticBaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
     id: int
     created_at: datetime
     updated_at: datetime
-
-    class Config:
-        from_attributes = True