base.php — AI Engine – The Chatbot, AI Framework & MCP for WordPress 1.9.2

ai-engine / classes / queries / base.php

ai-engine / classes / queries Last commit date

base.php 2 years ago embed.php 2 years ago function.php 2 years ago image.php 2 years ago parameter.php 2 years ago text.php 2 years ago transcribe.php 2 years ago

base.php

196 lines

1	<?php
2
3	class Meow_MWAI_Query_Base implements JsonSerializable {
4	public string $env = '';
5	public string $prompt = '';
6	public string $model = '';
7	public string $mode = '';
8	public ?string $session = null;
9	public int $maxResults = 1;
10	public ?string $service = null;
11	public ?string $botId = null;
12
13	// Functions
14	public array $functions = [];
15	public ?string $functionCall = null;
16
17	// OpenAI
18	public ?string $apiKey = null;
19
20	// Azure
21	public ?string $azureEndpoint = null;
22	public ?string $azureApiKey = null;
23	public ?string $azureDeployment = null;
24
25	public function __construct( $prompt = '' ) {
26	global $mwai_core;
27	$this->setPrompt( $prompt );
28	$this->session = $mwai_core->get_session_id();
29	}
30
31	#[\ReturnTypeWillChange]
32	public function jsonSerialize() {
33	return [
34	'class' => get_class( $this ),
35	'env' => $this->env,
36	'prompt' => $this->prompt,
37	'model' => $this->model,
38	'mode' => $this->mode,
39	'session' => $this->session,
40	'maxResults' => $this->maxResults
41	];
42	}
43
44	public function addFunction( Meow_MWAI_Query_Function $function ): void {
45	$this->functions[] = $function;
46	$this->functionCall = "auto";
47	}
48
49	public function setFunctions( array $functions ): void {
50	$this->functions = $functions;
51	$this->functionCall = "auto";
52	}
53
54	public function getFunctions(): array {
55	return $this->functions;
56	}
57
58	public function replace( $search, $replace ) {
59	$this->prompt = str_replace( $search, $replace, $this->prompt );
60	}
61
62	public function getLastPrompt(): string {
63	return $this->prompt;
64	}
65
66	/**
67	* The environment, like "chatbot", "imagesbot", "chatbot-007", "textwriter", etc...
68	* Used for statistics, mainly.
69	* @param string $env The environment.
70	*/
71	public function setEnv( string $env ): void {
72	$this->env = $env;
73	}
74
75	/**
76	* ID of the model to use.
77	* @param string $model ID of the model to use.
78	*/
79	public function setModel( string $model ) {
80	$this->model = $model;
81	}
82
83	/**
84	* The mode
85	* @param string $mode.
86	*/
87	public function setMode( string $mode ) {
88	$this->mode = $mode;
89	}
90
91	/**
92	* Given a prompt, the model will return one or more predicted completions.
93	* It can also return the probabilities of alternative tokens at each position.
94	* @param string $prompt The prompt to generate completions.
95	*/
96	public function setPrompt( string $prompt ) {
97	$this->prompt = $prompt;
98	}
99
100	public function getPrompt() {
101	return $this->prompt;
102	}
103
104	/**
105	* Similar to the prompt, but focus on the new/last message.
106	* Only used when the model has a chat mode (and only used in messages).
107	* With Meow_MWAI_Query_Base, this is the same as setPrompt.
108	* @param string $prompt The messages to generate completions.
109	*/
110	public function setNewMessage( string $newMessage ): void {
111	$this->setPrompt( $newMessage );
112	}
113
114	public function getLastMessage() {
115	return $this->getPrompt();
116	}
117
118	/**
119	* The API key to use.
120	* @param string $apiKey The API key.
121	*/
122	public function setApiKey( string $apiKey ) {
123	$this->apiKey = $apiKey;
124	}
125
126	/**
127	* The service to use.
128	* @param string $service The service.
129	*/
130	public function setService( string $service ) {
131	$this->service = $service;
132	}
133
134	/**
135	* The Azure endpoint to use.
136	* @param string $endpoint The endpoint.
137	*/
138	public function setAzureEndpoint( string $endpoint ) {
139	$this->azureEndpoint = $endpoint;
140	}
141
142	/**
143	* The Azure API key to use.
144	* @param string $apiKey The API key.
145	*/
146	public function setAzureApiKey( string $apiKey ) {
147	$this->azureApiKey = $apiKey;
148	}
149
150	/**
151	* The Azure deployment to use.
152	* @param string $deployment The deployment.
153	*/
154	public function setAzureDeployment( string $deployment ) {
155	$this->azureDeployment = $deployment;
156	}
157
158	/**
159	* The session ID to use.
160	* @param string $session The session ID.
161	*/
162	public function setSession( string $session ) {
163	$this->session = $session;
164	}
165
166	/**
167	* The bot ID to use.
168	* @param string $botId The bot ID.
169	*/
170	public function setBotId( string $botId ) {
171	$this->botId = $botId;
172	}
173
174	/**
175	* How many completions to generate for each prompt.
176	* Because this parameter generates many completions, it can quickly consume your token quota.
177	* Use carefully and ensure that you have reasonable settings for max_tokens and stop.
178	* @param float $maxResults Number of completions.
179	*/
180	public function setMaxResults( int $maxResults ) {
181	$this->maxResults = $maxResults;
182	}
183
184	// **
185	// * Check if everything is correct, otherwise fix it (like the max number of tokens).
186	// */
187	public function finalChecks() {
188	}
189
190	/*
191	* Get the JSON representation of the query.
192	*/
193	public function toJson() {
194	return json_encode( $this );
195	}
196	}