Skip to content

Commit 046137a

Browse files
Merge pull request #170 from forcedotcom/mz/jsonschema-callLLM
feat: add json_schema as parameter to call LLM
2 parents 4e44e80 + ea9be01 commit 046137a

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/types/llmService/LLMService.ts

+14-1
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,24 @@ export interface LLMServiceInterface {
1414
* @param engineeredPrompt - The prompt that has been engineered for the LLM.
1515
* @param promptId - The ID of the prompt (optional).
1616
* @param outputTokenLimit - The limit on the number of output tokens (optional).
17+
* @param options - The object of other options in the request (optional)
1718
* @returns A promise that resolves to the LLM's response as a string.
1819
*/
1920
callLLM(
2021
engineeredPrompt: string,
2122
promptId?: string,
22-
outputTokenLimit?: number
23+
outputTokenLimit?: number,
24+
options?: CallLLMOptions
2325
): Promise<string>;
2426
}
27+
28+
export type CallLLMOptions = {
29+
// Values to be added to the parameters property on the request.
30+
parameters?: {
31+
guided_json?: string; // TODO: define the type in detail
32+
};
33+
// Properties to be added to the body of the request.
34+
properties?: {
35+
stop_sequence?: string[]; // TODO: define the type in detail
36+
};
37+
};

0 commit comments

Comments
 (0)