Aasher commited on
Commit
f9f6b95
·
1 Parent(s): 4354e29

Add config for the agent

Browse files
Files changed (3) hide show
  1. pyproject.toml +5 -1
  2. src/axiom/config.py +30 -0
  3. uv.lock +0 -0
pyproject.toml CHANGED
@@ -4,4 +4,8 @@ version = "0.1.0"
4
  description = "An Advanced version of Axiom Agent"
5
  readme = "README.md"
6
  requires-python = ">=3.11"
7
- dependencies = []
 
 
 
 
 
4
  description = "An Advanced version of Axiom Agent"
5
  readme = "README.md"
6
  requires-python = ">=3.11"
7
+ dependencies = [
8
+ "chainlit>=2.5.5",
9
+ "openai-agents>=0.0.11",
10
+ "python-dotenv>=1.1.0",
11
+ ]
src/axiom/config.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dotenv import load_dotenv
3
+
4
+ from pydantic_settings import BaseSettings, SettingsConfigDict
5
+
6
+ _ = load_dotenv()
7
+
8
+ class Settings(BaseSettings):
9
+ model_config = SettingsConfigDict(
10
+ env_file=".env", extra="ignore", env_file_encoding="utf-8"
11
+ )
12
+ """
13
+ Configuration settings for the Axiom 2.0 Agent.
14
+ """
15
+ MODEL_PROVIDER: str = "google"
16
+ MODEL_API_KEY: str = os.getenv("GOOGLE_API_KEY")
17
+
18
+ AVAILABLE_MODELS: list[str] = [
19
+ "gemini-2.0-flash",
20
+ "gemini-2.0-flash-lite",
21
+ "gemini-2.0-flash-thinking-exp-1219",
22
+ "gemini-2.5-pro-exp-03-25",
23
+ "gemini-2.5-flash-preview-04-17",
24
+
25
+ ]
26
+ DEFAULT_MODEL: str = "gemini-2.0-flash"
27
+
28
+ MAX_DOCS_TOKEN_LIMIT: int = 20000 # Maximum tokens to retrieve from the documentations
29
+
30
+ settings = Settings()
uv.lock ADDED
The diff for this file is too large to render. See raw diff