fix: extra forbid

nobu007 · Sep 22, 2024 · 624ad59 · 624ad59
1 parent 067a598
commit 624ad59
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/codeinterpreterapi/thoughts/base.py b/src/codeinterpreterapi/thoughts/base.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 from textwrap import indent
-from pydantic import Field
 from typing import Any, Dict, List, Optional, Type
 
 from langchain_core.language_models import BaseLanguageModel
@@ -38,7 +37,7 @@ class MyToTChain(ToTChain):
     class Config:
         """Configuration for this pydantic object."""
 
-        extra = Field(default="forbid")
+        extra = 'forbid'
         arbitrary_types_allowed = True
 
     def initialize_thought_generator(self):