tako-sdk 1.0.1 → 1.0.3

package/dist/index.cjs

@@ -65,6 +65,10 @@ __export(index_exports, {
   AgentStreamEnvelopeFromJSONTyped: () => AgentStreamEnvelopeFromJSONTyped,
   AgentStreamEnvelopeToJSON: () => AgentStreamEnvelopeToJSON,
   AgentStreamEnvelopeToJSONTyped: () => AgentStreamEnvelopeToJSONTyped,
+  AgentUsageFromJSON: () => AgentUsageFromJSON,
+  AgentUsageFromJSONTyped: () => AgentUsageFromJSONTyped,
+  AgentUsageToJSON: () => AgentUsageToJSON,
+  AgentUsageToJSONTyped: () => AgentUsageToJSONTyped,
   AnswerResponseFromJSON: () => AnswerResponseFromJSON,
   AnswerResponseFromJSONTyped: () => AnswerResponseFromJSONTyped,
   AnswerResponseToJSON: () => AnswerResponseToJSON,
@@ -94,6 +98,14 @@ __export(index_exports, {
   CardSourcePrivateIndexFromJSONTyped: () => CardSourcePrivateIndexFromJSONTyped,
   CardSourcePrivateIndexToJSON: () => CardSourcePrivateIndexToJSON,
   CardSourcePrivateIndexToJSONTyped: () => CardSourcePrivateIndexToJSONTyped,
+  ClassifyRequestFromJSON: () => ClassifyRequestFromJSON,
+  ClassifyRequestFromJSONTyped: () => ClassifyRequestFromJSONTyped,
+  ClassifyRequestToJSON: () => ClassifyRequestToJSON,
+  ClassifyRequestToJSONTyped: () => ClassifyRequestToJSONTyped,
+  ClassifyResponseFromJSON: () => ClassifyResponseFromJSON,
+  ClassifyResponseFromJSONTyped: () => ClassifyResponseFromJSONTyped,
+  ClassifyResponseToJSON: () => ClassifyResponseToJSON,
+  ClassifyResponseToJSONTyped: () => ClassifyResponseToJSONTyped,
   ComponentConfigFromJSON: () => ComponentConfigFromJSON,
   ComponentConfigFromJSONTyped: () => ComponentConfigFromJSONTyped,
   ComponentConfigToJSON: () => ComponentConfigToJSON,
@@ -180,6 +192,10 @@ __export(index_exports, {
   OutputSettingsFromJSONTyped: () => OutputSettingsFromJSONTyped,
   OutputSettingsToJSON: () => OutputSettingsToJSON,
   OutputSettingsToJSONTyped: () => OutputSettingsToJSONTyped,
+  QueryClassificationFromJSON: () => QueryClassificationFromJSON,
+  QueryClassificationFromJSONTyped: () => QueryClassificationFromJSONTyped,
+  QueryClassificationToJSON: () => QueryClassificationToJSON,
+  QueryClassificationToJSONTyped: () => QueryClassificationToJSONTyped,
   ReasoningEventFromJSON: () => ReasoningEventFromJSON,
   ReasoningEventFromJSONTyped: () => ReasoningEventFromJSONTyped,
   ReasoningEventKindEnum: () => ReasoningEventKindEnum,
@@ -294,11 +310,14 @@ __export(index_exports, {
   instanceOfAgentRunRequest: () => instanceOfAgentRunRequest,
   instanceOfAgentRunStatus: () => instanceOfAgentRunStatus,
   instanceOfAgentStreamEnvelope: () => instanceOfAgentStreamEnvelope,
+  instanceOfAgentUsage: () => instanceOfAgentUsage,
   instanceOfAnswerResponse: () => instanceOfAnswerResponse,
   instanceOfBaseAPIError: () => instanceOfBaseAPIError,
   instanceOfCardSourceIndex: () => instanceOfCardSourceIndex,
   instanceOfCardSourceIndexSegment: () => instanceOfCardSourceIndexSegment,
   instanceOfCardSourcePrivateIndex: () => instanceOfCardSourcePrivateIndex,
+  instanceOfClassifyRequest: () => instanceOfClassifyRequest,
+  instanceOfClassifyResponse: () => instanceOfClassifyResponse,
   instanceOfComponentConfig: () => instanceOfComponentConfig,
   instanceOfComponentTypeEnum: () => instanceOfComponentTypeEnum,
   instanceOfContentFormat: () => instanceOfContentFormat,
@@ -318,6 +337,7 @@ __export(index_exports, {
   instanceOfKnowledgeCardSource: () => instanceOfKnowledgeCardSource,
   instanceOfKnowledgeCardSourceIndexesInner: () => instanceOfKnowledgeCardSourceIndexesInner,
   instanceOfOutputSettings: () => instanceOfOutputSettings,
+  instanceOfQueryClassification: () => instanceOfQueryClassification,
   instanceOfReasoningEvent: () => instanceOfReasoningEvent,
   instanceOfResultContent: () => instanceOfResultContent,
   instanceOfSearchEffortLevel: () => instanceOfSearchEffortLevel,
@@ -647,6 +667,33 @@ var TextApiResponse = class {
   }
 };
 
+// src/generated/models/AgentUsage.ts
+function instanceOfAgentUsage(value) {
+  return true;
+}
+function AgentUsageFromJSON(json) {
+  return AgentUsageFromJSONTyped(json, false);
+}
+function AgentUsageFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "usd_cost": json["usd_cost"] == null ? void 0 : json["usd_cost"]
+  };
+}
+function AgentUsageToJSON(json) {
+  return AgentUsageToJSONTyped(json, false);
+}
+function AgentUsageToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "usd_cost": value["usd_cost"]
+  };
+}
+
 // src/generated/models/KnowledgeCardMethodology.ts
 function instanceOfKnowledgeCardMethodology(value) {
   if (!("methodology_name" in value) || value["methodology_name"] === void 0) return false;
@@ -929,7 +976,8 @@ function TakoCardFromJSONTyped(json, ignoreDiscriminator) {
     "source_indexes": json["source_indexes"] == null ? void 0 : json["source_indexes"].map(KnowledgeCardSourceIndexesInnerFromJSON),
     "card_type": json["card_type"] == null ? void 0 : json["card_type"],
     "relevance": json["relevance"] == null ? void 0 : KnowledgeCardRelevanceFromJSON(json["relevance"]),
-    "content": json["content"] == null ? void 0 : ResultContentFromJSON(json["content"])
+    "content": json["content"] == null ? void 0 : ResultContentFromJSON(json["content"]),
+    "relevance_score": json["relevance_score"] == null ? void 0 : json["relevance_score"]
   };
 }
 function TakoCardToJSON(json) {
@@ -952,7 +1000,8 @@ function TakoCardToJSONTyped(value, ignoreDiscriminator = false) {
     "source_indexes": value["source_indexes"] == null ? void 0 : value["source_indexes"].map(KnowledgeCardSourceIndexesInnerToJSON),
     "card_type": value["card_type"],
     "relevance": KnowledgeCardRelevanceToJSON(value["relevance"]),
-    "content": ResultContentToJSON(value["content"])
+    "content": ResultContentToJSON(value["content"]),
+    "relevance_score": value["relevance_score"]
   };
 }
 
@@ -1030,6 +1079,105 @@ function AgentResultToJSONTyped(value, ignoreDiscriminator = false) {
   };
 }
 
+// src/generated/models/AgentEffortLevel.ts
+var AgentEffortLevel = {
+  Low: "low",
+  Medium: "medium"
+};
+function instanceOfAgentEffortLevel(value) {
+  for (const key in AgentEffortLevel) {
+    if (Object.prototype.hasOwnProperty.call(AgentEffortLevel, key)) {
+      if (AgentEffortLevel[key] === value) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function AgentEffortLevelFromJSON(json) {
+  return AgentEffortLevelFromJSONTyped(json, false);
+}
+function AgentEffortLevelFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+function AgentEffortLevelToJSON(value) {
+  return value;
+}
+function AgentEffortLevelToJSONTyped(value, ignoreDiscriminator) {
+  return value;
+}
+
+// src/generated/models/AgentOutputSettings.ts
+function instanceOfAgentOutputSettings(value) {
+  return true;
+}
+function AgentOutputSettingsFromJSON(json) {
+  return AgentOutputSettingsFromJSONTyped(json, false);
+}
+function AgentOutputSettingsFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "image_dark_mode": json["image_dark_mode"] == null ? void 0 : json["image_dark_mode"]
+  };
+}
+function AgentOutputSettingsToJSON(json) {
+  return AgentOutputSettingsToJSONTyped(json, false);
+}
+function AgentOutputSettingsToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "image_dark_mode": value["image_dark_mode"]
+  };
+}
+
+// src/generated/models/AgentRunRequest.ts
+var AgentRunRequestSourceIndexesEnum = {
+  Data: "data",
+  Web: "web"
+};
+function instanceOfAgentRunRequest(value) {
+  if (!("query" in value) || value["query"] === void 0) return false;
+  return true;
+}
+function AgentRunRequestFromJSON(json) {
+  return AgentRunRequestFromJSONTyped(json, false);
+}
+function AgentRunRequestFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "query": json["query"],
+    "thread_id": json["thread_id"] == null ? void 0 : json["thread_id"],
+    "effort": json["effort"] == null ? void 0 : AgentEffortLevelFromJSON(json["effort"]),
+    "source_indexes": json["source_indexes"] == null ? void 0 : json["source_indexes"],
+    "locale": json["locale"] == null ? void 0 : json["locale"],
+    "timezone": json["timezone"] == null ? void 0 : json["timezone"],
+    "output_settings": json["output_settings"] == null ? void 0 : AgentOutputSettingsFromJSON(json["output_settings"])
+  };
+}
+function AgentRunRequestToJSON(json) {
+  return AgentRunRequestToJSONTyped(json, false);
+}
+function AgentRunRequestToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "query": value["query"],
+    "thread_id": value["thread_id"],
+    "effort": AgentEffortLevelToJSON(value["effort"]),
+    "source_indexes": value["source_indexes"],
+    "locale": value["locale"],
+    "timezone": value["timezone"],
+    "output_settings": AgentOutputSettingsToJSON(value["output_settings"])
+  };
+}
+
 // src/generated/models/AgentRunStatus.ts
 var AgentRunStatus = {
   Queued: "queued",
@@ -1116,7 +1264,9 @@ function AgentRunFromJSONTyped(json, ignoreDiscriminator) {
     "created_at": json["created_at"],
     "completed_at": json["completed_at"] == null ? void 0 : json["completed_at"],
     "result": json["result"] == null ? void 0 : AgentResultFromJSON(json["result"]),
-    "error": json["error"] == null ? void 0 : ErrorObjectFromJSON(json["error"])
+    "error": json["error"] == null ? void 0 : ErrorObjectFromJSON(json["error"]),
+    "usage": json["usage"] == null ? void 0 : AgentUsageFromJSON(json["usage"]),
+    "request": json["request"] == null ? void 0 : AgentRunRequestFromJSON(json["request"])
   };
 }
 function AgentRunToJSON(json) {
@@ -1134,105 +1284,9 @@ function AgentRunToJSONTyped(value, ignoreDiscriminator = false) {
     "created_at": value["created_at"],
     "completed_at": value["completed_at"],
     "result": AgentResultToJSON(value["result"]),
-    "error": ErrorObjectToJSON(value["error"])
-  };
-}
-
-// src/generated/models/AgentEffortLevel.ts
-var AgentEffortLevel = {
-  Medium: "medium"
-};
-function instanceOfAgentEffortLevel(value) {
-  for (const key in AgentEffortLevel) {
-    if (Object.prototype.hasOwnProperty.call(AgentEffortLevel, key)) {
-      if (AgentEffortLevel[key] === value) {
-        return true;
-      }
-    }
-  }
-  return false;
-}
-function AgentEffortLevelFromJSON(json) {
-  return AgentEffortLevelFromJSONTyped(json, false);
-}
-function AgentEffortLevelFromJSONTyped(json, ignoreDiscriminator) {
-  return json;
-}
-function AgentEffortLevelToJSON(value) {
-  return value;
-}
-function AgentEffortLevelToJSONTyped(value, ignoreDiscriminator) {
-  return value;
-}
-
-// src/generated/models/AgentOutputSettings.ts
-function instanceOfAgentOutputSettings(value) {
-  return true;
-}
-function AgentOutputSettingsFromJSON(json) {
-  return AgentOutputSettingsFromJSONTyped(json, false);
-}
-function AgentOutputSettingsFromJSONTyped(json, ignoreDiscriminator) {
-  if (json == null) {
-    return json;
-  }
-  return {
-    "image_dark_mode": json["image_dark_mode"] == null ? void 0 : json["image_dark_mode"]
-  };
-}
-function AgentOutputSettingsToJSON(json) {
-  return AgentOutputSettingsToJSONTyped(json, false);
-}
-function AgentOutputSettingsToJSONTyped(value, ignoreDiscriminator = false) {
-  if (value == null) {
-    return value;
-  }
-  return {
-    "image_dark_mode": value["image_dark_mode"]
-  };
-}
-
-// src/generated/models/AgentRunRequest.ts
-var AgentRunRequestSourceIndexesEnum = {
-  Tako: "tako",
-  Web: "web"
-};
-function instanceOfAgentRunRequest(value) {
-  if (!("query" in value) || value["query"] === void 0) return false;
-  return true;
-}
-function AgentRunRequestFromJSON(json) {
-  return AgentRunRequestFromJSONTyped(json, false);
-}
-function AgentRunRequestFromJSONTyped(json, ignoreDiscriminator) {
-  if (json == null) {
-    return json;
-  }
-  return {
-    "query": json["query"],
-    "thread_id": json["thread_id"] == null ? void 0 : json["thread_id"],
-    "effort": json["effort"] == null ? void 0 : AgentEffortLevelFromJSON(json["effort"]),
-    "source_indexes": json["source_indexes"] == null ? void 0 : json["source_indexes"],
-    "locale": json["locale"] == null ? void 0 : json["locale"],
-    "timezone": json["timezone"] == null ? void 0 : json["timezone"],
-    "output_settings": json["output_settings"] == null ? void 0 : AgentOutputSettingsFromJSON(json["output_settings"])
-  };
-}
-function AgentRunRequestToJSON(json) {
-  return AgentRunRequestToJSONTyped(json, false);
-}
-function AgentRunRequestToJSONTyped(value, ignoreDiscriminator = false) {
-  if (value == null) {
-    return value;
-  }
-  return {
-    "query": value["query"],
-    "thread_id": value["thread_id"],
-    "effort": AgentEffortLevelToJSON(value["effort"]),
-    "source_indexes": value["source_indexes"],
-    "locale": value["locale"],
-    "timezone": value["timezone"],
-    "output_settings": AgentOutputSettingsToJSON(value["output_settings"])
+    "error": ErrorObjectToJSON(value["error"]),
+    "usage": AgentUsageToJSON(value["usage"]),
+    "request": AgentRunRequestToJSON(value["request"])
   };
 }
 
@@ -1357,6 +1411,95 @@ function AnswerResponseToJSONTyped(value, ignoreDiscriminator = false) {
   };
 }
 
+// src/generated/models/ClassifyRequest.ts
+function instanceOfClassifyRequest(value) {
+  if (!("queries" in value) || value["queries"] === void 0) return false;
+  return true;
+}
+function ClassifyRequestFromJSON(json) {
+  return ClassifyRequestFromJSONTyped(json, false);
+}
+function ClassifyRequestFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "queries": json["queries"]
+  };
+}
+function ClassifyRequestToJSON(json) {
+  return ClassifyRequestToJSONTyped(json, false);
+}
+function ClassifyRequestToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "queries": value["queries"]
+  };
+}
+
+// src/generated/models/QueryClassification.ts
+function instanceOfQueryClassification(value) {
+  if (!("query" in value) || value["query"] === void 0) return false;
+  if (!("score" in value) || value["score"] === void 0) return false;
+  if (!("keep" in value) || value["keep"] === void 0) return false;
+  return true;
+}
+function QueryClassificationFromJSON(json) {
+  return QueryClassificationFromJSONTyped(json, false);
+}
+function QueryClassificationFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "query": json["query"],
+    "score": json["score"],
+    "keep": json["keep"]
+  };
+}
+function QueryClassificationToJSON(json) {
+  return QueryClassificationToJSONTyped(json, false);
+}
+function QueryClassificationToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "query": value["query"],
+    "score": value["score"],
+    "keep": value["keep"]
+  };
+}
+
+// src/generated/models/ClassifyResponse.ts
+function instanceOfClassifyResponse(value) {
+  return true;
+}
+function ClassifyResponseFromJSON(json) {
+  return ClassifyResponseFromJSONTyped(json, false);
+}
+function ClassifyResponseFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    "results": json["results"] == null ? void 0 : json["results"].map(QueryClassificationFromJSON)
+  };
+}
+function ClassifyResponseToJSON(json) {
+  return ClassifyResponseToJSONTyped(json, false);
+}
+function ClassifyResponseToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    "results": value["results"] == null ? void 0 : value["results"].map(QueryClassificationToJSON)
+  };
+}
+
 // src/generated/models/ContentsDeliveryMode.ts
 var ContentsDeliveryMode = {
   Url: "url",
@@ -1713,7 +1856,8 @@ function KnowledgeCardToJSONTyped(value, ignoreDiscriminator = false) {
 // src/generated/models/SearchEffortLevel.ts
 var SearchEffortLevel = {
   Fast: "fast",
-  Instant: "instant"
+  Instant: "instant",
+  Deep: "deep"
 };
 function instanceOfSearchEffortLevel(value) {
   for (const key in SearchEffortLevel) {
@@ -1839,7 +1983,7 @@ function SourcesFromJSONTyped(json, ignoreDiscriminator) {
     return json;
   }
   return {
-    "tako": json["tako"] == null ? void 0 : TakoSourceSettingsFromJSON(json["tako"]),
+    "data": json["data"] == null ? void 0 : TakoSourceSettingsFromJSON(json["data"]),
     "web": json["web"] == null ? void 0 : SourceSettingsFromJSON(json["web"])
   };
 }
@@ -1851,7 +1995,7 @@ function SourcesToJSONTyped(value, ignoreDiscriminator = false) {
     return value;
   }
   return {
-    "tako": TakoSourceSettingsToJSON(value["tako"]),
+    "data": TakoSourceSettingsToJSON(value["data"]),
     "web": SourceSettingsToJSON(value["web"])
   };
 }
@@ -1968,6 +2112,42 @@ var TakoApi = class extends BaseAPI {
     const response = await this.answerRaw(requestParameters, initOverrides);
     return await response.value();
   }
+  /**
+   * Creates request options for classify without sending the request
+   */
+  async classifyRequestOpts(requestParameters) {
+    const queryParameters = {};
+    const headerParameters = {};
+    headerParameters["Content-Type"] = "application/json";
+    if (this.configuration && this.configuration.apiKey) {
+      headerParameters["X-API-Key"] = await this.configuration.apiKey("X-API-Key");
+    }
+    let urlPath = `/v1/classify`;
+    return {
+      path: urlPath,
+      method: "POST",
+      headers: headerParameters,
+      query: queryParameters,
+      body: ClassifyRequestToJSON(requestParameters["classifyRequest"])
+    };
+  }
+  /**
+   * Score a batch of queries for whether Tako search will return a result. Returns a probability and a recall-skewed keep flag per query. Free; intended for pre-filtering before /v3/search.
+   * Classify queries
+   */
+  async classifyRaw(requestParameters, initOverrides) {
+    const requestOptions = await this.classifyRequestOpts(requestParameters);
+    const response = await this.request(requestOptions, initOverrides);
+    return new JSONApiResponse(response, (jsonValue) => ClassifyResponseFromJSON(jsonValue));
+  }
+  /**
+   * Score a batch of queries for whether Tako search will return a result. Returns a probability and a recall-skewed keep flag per query. Free; intended for pre-filtering before /v3/search.
+   * Classify queries
+   */
+  async classify(requestParameters = {}, initOverrides) {
+    const response = await this.classifyRaw(requestParameters, initOverrides);
+    return await response.value();
+  }
   /**
    * Creates request options for contents without sending the request
    */
@@ -2086,7 +2266,8 @@ var APIErrorType = {
   PaymentRequired: "PAYMENT_REQUIRED",
   RequestTimeout: "REQUEST_TIMEOUT",
   Forbidden: "FORBIDDEN",
-  NotFound: "NOT_FOUND"
+  NotFound: "NOT_FOUND",
+  ServiceUnavailable: "SERVICE_UNAVAILABLE"
 };
 function instanceOfAPIErrorType(value) {
   for (const key in APIErrorType) {
@@ -3117,6 +3298,10 @@ var Tako = class {
   AgentStreamEnvelopeFromJSONTyped,
   AgentStreamEnvelopeToJSON,
   AgentStreamEnvelopeToJSONTyped,
+  AgentUsageFromJSON,
+  AgentUsageFromJSONTyped,
+  AgentUsageToJSON,
+  AgentUsageToJSONTyped,
   AnswerResponseFromJSON,
   AnswerResponseFromJSONTyped,
   AnswerResponseToJSON,
@@ -3146,6 +3331,14 @@ var Tako = class {
   CardSourcePrivateIndexFromJSONTyped,
   CardSourcePrivateIndexToJSON,
   CardSourcePrivateIndexToJSONTyped,
+  ClassifyRequestFromJSON,
+  ClassifyRequestFromJSONTyped,
+  ClassifyRequestToJSON,
+  ClassifyRequestToJSONTyped,
+  ClassifyResponseFromJSON,
+  ClassifyResponseFromJSONTyped,
+  ClassifyResponseToJSON,
+  ClassifyResponseToJSONTyped,
   ComponentConfigFromJSON,
   ComponentConfigFromJSONTyped,
   ComponentConfigToJSON,
@@ -3232,6 +3425,10 @@ var Tako = class {
   OutputSettingsFromJSONTyped,
   OutputSettingsToJSON,
   OutputSettingsToJSONTyped,
+  QueryClassificationFromJSON,
+  QueryClassificationFromJSONTyped,
+  QueryClassificationToJSON,
+  QueryClassificationToJSONTyped,
   ReasoningEventFromJSON,
   ReasoningEventFromJSONTyped,
   ReasoningEventKindEnum,
@@ -3346,11 +3543,14 @@ var Tako = class {
   instanceOfAgentRunRequest,
   instanceOfAgentRunStatus,
   instanceOfAgentStreamEnvelope,
+  instanceOfAgentUsage,
   instanceOfAnswerResponse,
   instanceOfBaseAPIError,
   instanceOfCardSourceIndex,
   instanceOfCardSourceIndexSegment,
   instanceOfCardSourcePrivateIndex,
+  instanceOfClassifyRequest,
+  instanceOfClassifyResponse,
   instanceOfComponentConfig,
   instanceOfComponentTypeEnum,
   instanceOfContentFormat,
@@ -3370,6 +3570,7 @@ var Tako = class {
   instanceOfKnowledgeCardSource,
   instanceOfKnowledgeCardSourceIndexesInner,
   instanceOfOutputSettings,
+  instanceOfQueryClassification,
   instanceOfReasoningEvent,
   instanceOfResultContent,
   instanceOfSearchEffortLevel,