Pods

src/task.rs

@@ -174,6 +174,15 @@ pub struct FileRange {
     pub line_end: u32,
 }
 
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ConnectorConfig {
+    pub api_key: String,
+    pub base_url: String,
+    pub model: String,
+    pub max_tokens: Option<i32>,
+    pub temperature: Option<f32>,
+}
+
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct AgentTaskSpec {
     pub task_id: Uuid,
@@ -196,6 +205,10 @@ pub struct AgentTaskSpec {
     pub suggested_test_files: Vec<String>,
     #[serde(default)]
     pub project_id: Option<Uuid>,
+    #[serde(default)]
+    pub connector_config: Option<ConnectorConfig>,
+    #[serde(default)]
+    pub tools_json: Option<String>,
 }
 
 impl AgentTaskSpec {
@@ -224,6 +237,8 @@ impl AgentTaskSpec {
             agent_type: None,
             suggested_test_files: Vec::new(),
             project_id: None,
+            connector_config: None,
+            tools_json: None,
         }
     }
 }