← All changes | includes/class-mxchat-utils.php +39 -58 2.0.6 → 1.6.2 View file →

		@@ -67,68 +67,49 @@
67	67	// error_log('Embedding generation failed for content from ' . $source_url);
68	68	}
69	69	}
70	70
71		-/**
72		-* Generate an embedding for the given text using the specified API key.
73		-*
74		-* @param string $text The text to be embedded.
75		-* @param string $api_key The API key used for generating embeddings.
76		-* @return array|null The embedding vector or null on failure.
77		-*/
78		-private static function generate_embedding($text, $api_key) {
79		- // Get options and selected model
80		- $options = get_option('mxchat_options');
81		- $selected_model = $options['embedding_model'] ?? 'text-embedding-ada-002';
82		-
83		- // Determine endpoint and API key based on model
84		- if (strpos($selected_model, 'voyage') === 0) {
85		- $endpoint = 'https://api.voyageai.com/v1/embeddings';
86		- $api_key = $options['voyage_api_key'] ?? '';
87		- } else {
88		- $endpoint = 'https://api.openai.com/v1/embeddings';
89		- // Use the passed API key for OpenAI
90		- }
	71	+ /**
	72	+ * Generate an embedding for the given text using the specified API key.
	73	+ *
	74	+ * @param string $text The text to be embedded.
	75	+ * @param string $api_key The API key used for generating embeddings.
	76	+ * @return array|null The embedding vector or null on failure.
	77	+ */
	78	+ private static function generate_embedding($text, $api_key) {
	79	+ $endpoint = 'https://api.openai.com/v1/embeddings';
91	80
92		- // Prepare request body
93		- $request_body = [
94		- 'input' => $text,
95		- 'model' => $selected_model
96		- ];
	81	+ $body = wp_json_encode([
	82	+ 'input' => $text,
	83	+ 'model' => 'text-embedding-ada-002'
	84	+ ]);
97	85
98		- // Add output_dimension for voyage-3-large
99		- if ($selected_model === 'voyage-3-large') {
100		- $request_body['output_dimension'] = 2048;
101		- }
	86	+ $args = [
	87	+ 'body' => $body,
	88	+ 'headers' => [
	89	+ 'Content-Type' => 'application/json',
	90	+ 'Authorization' => 'Bearer ' . $api_key,
	91	+ ],
	92	+ 'timeout' => 60,
	93	+ 'redirection' => 5,
	94	+ 'blocking' => true,
	95	+ 'httpversion' => '1.0',
	96	+ 'sslverify' => true,
	97	+ ];
102	98
103		- $args = [
104		- 'body' => wp_json_encode($request_body),
105		- 'headers' => [
106		- 'Content-Type' => 'application/json',
107		- 'Authorization' => 'Bearer ' . $api_key,
108		- ],
109		- 'timeout' => 60,
110		- 'redirection' => 5,
111		- 'blocking' => true,
112		- 'httpversion' => '1.0',
113		- 'sslverify' => true,
114		- ];
	99	+ $response = wp_remote_post($endpoint, $args);
115	100
116		- $response = wp_remote_post($endpoint, $args);
117		-
118		- if (is_wp_error($response)) {
119		- error_log('Error generating embedding: ' . $response->get_error_message());
120		- return null;
121		- }
	101	+ if (is_wp_error($response)) {
	102	+ // error_log('Error generating embedding: ' . $response->get_error_message());
	103	+ return null;
	104	+ }
122	105
123		- $response_body = json_decode(wp_remote_retrieve_body($response), true);
124		-
125		- if (isset($response_body['data'][0]['embedding']) && is_array($response_body['data'][0]['embedding'])) {
126		- return $response_body['data'][0]['embedding'];
127		- } else {
128		- error_log('Invalid response received from embedding API: ' . wp_json_encode($response_body));
129		- return null;
130		- }
131		-}
132		-
133		-
	106	+ $response_body = json_decode(wp_remote_retrieve_body($response), true);
	107	+
	108	+ if (isset($response_body['data'][0]['embedding']) && is_array($response_body['data'][0]['embedding'])) {
	109	+ return $response_body['data'][0]['embedding'];
	110	+ } else {
	111	+ // error_log('Invalid response received from embedding API.');
	112	+ return null;
	113	+ }
	114	+ }
134	115	}