Skip to content

Commit

Permalink
fix: silence pydantic warning about protected namespace (#1116)
Browse files Browse the repository at this point in the history
  • Loading branch information
cpacker authored Mar 8, 2024
1 parent 701b6eb commit 51c6d00
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion memgpt/models/pydantic_models.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from typing import List, Optional, Dict, Literal
from pydantic import BaseModel, Field, Json
from pydantic import BaseModel, Field, Json, ConfigDict
import uuid
from datetime import datetime
from sqlmodel import Field, SQLModel
Expand All @@ -15,6 +15,9 @@ class LLMConfigModel(BaseModel):
model_wrapper: Optional[str] = None
context_window: Optional[int] = None

# FIXME hack to silence pydantic protected namespace warning
model_config = ConfigDict(protected_namespaces=())


class EmbeddingConfigModel(BaseModel):
embedding_endpoint_type: Optional[str] = "openai"
Expand Down

0 comments on commit 51c6d00

Please sign in to comment.