pi9605 commited on
Commit
4df57fe
·
1 Parent(s): abd2333

updated inference.py

Browse files
Files changed (6) hide show
  1. LICENSE +21 -0
  2. inference.py +1 -1
  3. models.py +1 -1
  4. openenv.yaml +1 -1
  5. server/app.py +1 -1
  6. server/config.py +1 -1
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2026 subhdotsol
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
inference.py CHANGED
@@ -10,7 +10,7 @@ logging.basicConfig(level=logging.INFO)
10
  logger = logging.getLogger(__name__)
11
 
12
  # Backend server
13
- SERVER_URL = "https://rayugacodes-breach-os.hf.space"
14
 
15
  # Attacker LLM (Configured to Groq)
16
  client = OpenAI(
 
10
  logger = logging.getLogger(__name__)
11
 
12
  # Backend server
13
+ SERVER_URL = "http://localhost:8000"
14
 
15
  # Attacker LLM (Configured to Groq)
16
  client = OpenAI(
models.py CHANGED
@@ -1,5 +1,5 @@
1
  """
2
- Shared Pydantic models and Enums for the RedTeamOS environment.
3
  These define the API contract between the server, environment, and clients.
4
  """
5
  from pydantic import BaseModel, Field, field_validator
 
1
  """
2
+ Shared Pydantic models and Enums for the BreachOS environment.
3
  These define the API contract between the server, environment, and clients.
4
  """
5
  from pydantic import BaseModel, Field, field_validator
openenv.yaml CHANGED
@@ -1,4 +1,4 @@
1
- name: RedTeamOS
2
  version: "0.1.0"
3
  description: "AI Red-Teaming Environment for Safety Research"
4
 
 
1
+ name: BreachOS
2
  version: "0.1.0"
3
  description: "AI Red-Teaming Environment for Safety Research"
4
 
server/app.py CHANGED
@@ -26,7 +26,7 @@ async def lifespan(app: FastAPI):
26
  yield
27
 
28
  app = FastAPI(
29
- title = "RedTeamOS",
30
  version = "0.1.0",
31
  lifespan = lifespan,
32
  )
 
26
  yield
27
 
28
  app = FastAPI(
29
+ title = "BreachOS",
30
  version = "0.1.0",
31
  lifespan = lifespan,
32
  )
server/config.py CHANGED
@@ -16,7 +16,7 @@ class Settings(BaseSettings):
16
  llm_max_retries: int = 3
17
 
18
  class Config:
19
- env_file = ".env"
20
  env_file_encoding = "utf-8"
21
  extra = "ignore"
22
 
 
16
  llm_max_retries: int = 3
17
 
18
  class Config:
19
+ env_prefix = "BREACHOS_"
20
  env_file_encoding = "utf-8"
21
  extra = "ignore"
22