Skip to content

Commit

Permalink
clean
Browse files Browse the repository at this point in the history
  • Loading branch information
Mustafa Kerem Kurban committed Oct 3, 2024
1 parent c24c6fd commit 58be908
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 20 deletions.
16 changes: 1 addition & 15 deletions src/neuroagent/tools/bluenaas_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
Expand All @@ -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):
Expand All @@ -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"
Expand Down
5 changes: 0 additions & 5 deletions src/neuroagent/tools/get_me_model_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand Down

0 comments on commit 58be908

Please sign in to comment.