From 58be908468a594037851c588bd10f2dd36a88446 Mon Sep 17 00:00:00 2001
From: Mustafa Kerem Kurban <kurban@bb-g306hl04xd.epfl.ch>
Date: Thu, 3 Oct 2024 20:08:26 +0200
Subject: [PATCH] clean

---
 src/neuroagent/tools/bluenaas_tool.py     | 16 +---------------
 src/neuroagent/tools/get_me_model_tool.py |  5 -----
 2 files changed, 1 insertion(+), 20 deletions(-)

diff --git a/src/neuroagent/tools/bluenaas_tool.py b/src/neuroagent/tools/bluenaas_tool.py
index 7feff3e..ccedb20 100644
--- a/src/neuroagent/tools/bluenaas_tool.py
+++ b/src/neuroagent/tools/bluenaas_tool.py
@@ -12,12 +12,6 @@
 logger = logging.getLogger(__name__)
 
 #TODO : since bluenaaas has multiple endpoints for synapse generation ,nexus query, simulation, how should we handle those?
-# class SynapseSimulationConfig(BaseModel):
-#     id: str
-#     delay: int
-#     duration: int
-#     frequency: PositiveInt
-#     weightScalar: int
 
 class SimulationStimulusConfig(BaseModel):
     stimulusType: Literal["current_clamp", "voltage_clamp", "conductance"]
@@ -40,9 +34,6 @@ class SimulationConditionsConfig(BaseModel):
     time_step: float # usually 0.025 ms
     seed: int # can be any nubmer
 
-# class SimulationWithSynapseBody(BaseModel):
-#     directCurrentConfig: CurrentInjectionConfig
-#     synapseConfigs: list[SynapseSimulationConfig]
 
 
 class InputBlueNaaS(BaseModel):
@@ -54,12 +45,7 @@ class InputBlueNaaS(BaseModel):
             " fetched using the 'get-me-model-tool'."
         )
     )
-    # synapses: list[SynapseSimulationConfig] = Field(
-    #     description=(
-    #         "List of synapse configurations. Each synapse configuration includes the"
-    #         " synapse ID, delay, duration, frequency, and weight scalar."
-    #     )
-    # )
+
     currentInjection: CurrentInjectionConfig = Field(
         description=(
             "Configuration for current injection. Includes the target section to inject"
diff --git a/src/neuroagent/tools/get_me_model_tool.py b/src/neuroagent/tools/get_me_model_tool.py
index 7b267cf..f3c8498 100644
--- a/src/neuroagent/tools/get_me_model_tool.py
+++ b/src/neuroagent/tools/get_me_model_tool.py
@@ -14,11 +14,6 @@
 
 logger = logging.getLogger(__name__)
 
-# settings = Settings()
-# langsmith_api_key = settings.langsmith.api_key.get_secret_value()
-# langsmith_project = settings.langsmith.project
-# langsmith_endpoint = settings.langsmith.endpoint
-
 
 class InputGetMEModel(BaseModel):
     """Inputs of the knowledge graph API."""