3 files changed, 13 insertions, 2 deletions
diff --git a/extension/schema/ContinueConfig.d.ts b/extension/schema/ContinueConfig.d.ts
index f4c62191..5341056f 100644
--- a/extension/schema/ContinueConfig.d.ts
+++ b/extension/schema/ContinueConfig.d.ts
@@ -141,10 +141,17 @@ export interface Policy {
  */
 export interface ContextProvider {
   title: Title;
+  sdk?: ContinueSDK1;
   selected_items?: SelectedItems;
   [k: string]: unknown;
 }
 /**
+ * To avoid circular imports
+ */
+export interface ContinueSDK1 {
+  [k: string]: unknown;
+}
+/**
  * A ContextItem is a single item that is stored in the ContextManager.
  */
 export interface ContextItem {
diff --git a/extension/schema/FullState.d.ts b/extension/schema/FullState.d.ts
index a999ce95..a847a608 100644
--- a/extension/schema/FullState.d.ts
+++ b/extension/schema/FullState.d.ts
@@ -41,6 +41,7 @@ export type SelectedContextItems = ContextItem[];
 export type SessionId = string;
 export type Title = string;
 export type DateCreated = string;
+export type WorkspaceDirectory = string;
 export type SystemMessage1 = string;
 export type Temperature = number;
 
@@ -142,11 +143,12 @@ export interface SessionInfo {
   session_id: SessionId;
   title: Title;
   date_created: DateCreated;
+  workspace_directory?: WorkspaceDirectory;
   [k: string]: unknown;
 }
 export interface ContinueConfig {
-  system_message: SystemMessage1;
-  temperature: Temperature;
+  system_message?: SystemMessage1;
+  temperature?: Temperature;
   [k: string]: unknown;
 }
 export interface SavedContextGroups {
diff --git a/extension/schema/SessionInfo.d.ts b/extension/schema/SessionInfo.d.ts
index ac94e76d..80e3e527 100644
--- a/extension/schema/SessionInfo.d.ts
+++ b/extension/schema/SessionInfo.d.ts
@@ -9,10 +9,12 @@ export type SessionInfo = SessionInfo1;
 export type SessionId = string;
 export type Title = string;
 export type DateCreated = string;
+export type WorkspaceDirectory = string;
 
 export interface SessionInfo1 {
   session_id: SessionId;
   title: Title;
   date_created: DateCreated;
+  workspace_directory?: WorkspaceDirectory;
   [k: string]: unknown;
 }