Merge pull request #463 from Codium-ai/tr/more_protections

minor fix
This commit is contained in:
mrT23 2023-11-19 07:45:39 -08:00 committed by GitHub
commit 26cc26129c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -8,6 +8,7 @@ import pinecone
from pinecone_datasets import Dataset, DatasetMetadata
from pydantic import BaseModel, Field
from pr_agent.algo import MAX_TOKENS
from pr_agent.algo.token_handler import TokenHandler
from pr_agent.algo.utils import get_max_tokens
from pr_agent.config_loader import get_settings