1

keyBoard/Class/AiTalk/VM/AiVM.m

@@ -6,7 +6,250 @@
 //
 
 #import "AiVM.h"
+#import "KBAPI.h"
+#import "KBNetworkManager.h"
+#import <MJExtension/MJExtension.h>
+
+@implementation KBAiSyncData
+
+- (void)setAudioBase64:(NSString *)audioBase64 {
+  if (![audioBase64 isKindOfClass:[NSString class]]) {
+    _audioBase64 = nil;
+    self.audioData = nil;
+    return;
+  }
+  _audioBase64 = [audioBase64 copy];
+  if (_audioBase64.length == 0) {
+    self.audioData = nil;
+    return;
+  }
+
+  NSString *cleanBase64 = _audioBase64;
+  NSRange commaRange = [cleanBase64 rangeOfString:@","];
+  if ([cleanBase64 hasPrefix:@"data:"] && commaRange.location != NSNotFound) {
+    cleanBase64 = [cleanBase64 substringFromIndex:commaRange.location + 1];
+  }
+
+  self.audioData = [[NSData alloc]
+      initWithBase64EncodedString:cleanBase64
+                          options:NSDataBase64DecodingIgnoreUnknownCharacters];
+}
+
+@end
+
+@implementation KBAiSyncResponse
+@end
+
+@implementation KBAiMessageData
+@end
+
+@implementation KBAiMessageResponse
+@end
 
 @implementation AiVM
 
+- (void)syncChatWithTranscript:(NSString *)transcript
+                    completion:(AiVMSyncCompletion)completion {
+  if (transcript.length == 0) {
+    NSError *error = [NSError
+        errorWithDomain:@"AiVM"
+                   code:-1
+               userInfo:@{NSLocalizedDescriptionKey : @"transcript is empty"}];
+    if (completion) {
+      completion(nil, error);
+    }
+    return;
+  }
+
+  NSDictionary *params = @{ @"transcript" : transcript ?: @"" };
+  CFAbsoluteTime startTime = CFAbsoluteTimeGetCurrent();
+  NSLog(@"[AiVM] /chat/sync request: %@", params);
+  [[KBNetworkManager shared]
+            POST:API_AI_CHAT_SYNC
+        jsonBody:params
+         headers:nil
+autoShowBusinessError:NO
+      completion:^(NSDictionary *_Nullable json,
+                   NSURLResponse *_Nullable response,
+                   NSError *_Nullable error) {
+        CFAbsoluteTime elapsed =
+            (CFAbsoluteTimeGetCurrent() - startTime) * 1000.0;
+        if (error) {
+          NSLog(@"[AiVM] /chat/sync failed: %@",
+                error.localizedDescription ?: @"");
+          NSLog(@"[AiVM] /chat/sync duration: %.0f ms", elapsed);
+          if (completion) {
+            completion(nil, error);
+          }
+          return;
+        }
+      NSLog(@"[AiVM] /chat/sync response received");
+      NSLog(@"[AiVM] /chat/sync duration: %.0f ms", elapsed);
+        KBAiSyncResponse *model =
+            [KBAiSyncResponse mj_objectWithKeyValues:json];
+        
+        if (completion) {
+          completion(model, nil);
+        }
+      }];
+}
+
+- (void)requestChatMessageWithContent:(NSString *)content
+                           completion:(AiVMMessageCompletion)completion {
+  if (content.length == 0) {
+    NSError *error = [NSError
+        errorWithDomain:@"AiVM"
+                   code:-1
+               userInfo:@{NSLocalizedDescriptionKey : @"content is empty"}];
+    if (completion) {
+      completion(nil, error);
+    }
+    return;
+  }
+
+  NSString *encodedContent =
+      [content stringByAddingPercentEncodingWithAllowedCharacters:
+                   [NSCharacterSet URLQueryAllowedCharacterSet]];
+  NSString *path = [NSString
+      stringWithFormat:@"%@?content=%@", API_AI_CHAT_MESSAGE,
+                       encodedContent ?: @""];
+  NSDictionary *params = @{ @"content" : content ?: @"" };
+  [[KBNetworkManager shared]
+            POST:path
+        jsonBody:params
+         headers:nil
+autoShowBusinessError:NO
+      completion:^(NSDictionary *_Nullable json,
+                   NSURLResponse *_Nullable response,
+                   NSError *_Nullable error) {
+        if (error) {
+          if (completion) {
+            completion(nil, error);
+          }
+          return;
+        }
+
+        KBAiMessageResponse *model =
+            [KBAiMessageResponse mj_objectWithKeyValues:json];
+        if (completion) {
+          completion(model, nil);
+        }
+      }];
+}
+
+- (void)requestElevenLabsSpeechWithText:(NSString *)text
+                                voiceId:(NSString *)voiceId
+                                 apiKey:(NSString *)apiKey
+                            outputFormat:(NSString *)outputFormat
+                                 modelId:(NSString *)modelId
+                              completion:(AiVMElevenLabsCompletion)completion {
+  if (text.length == 0 || voiceId.length == 0 || apiKey.length == 0) {
+    NSError *error = [NSError
+        errorWithDomain:@"AiVM"
+                   code:-1
+               userInfo:@{NSLocalizedDescriptionKey : @"invalid parameters"}];
+    if (completion) {
+      completion(nil, error);
+    }
+    return;
+  }
+
+  NSString *format = outputFormat.length > 0 ? outputFormat : @"mp3_44100_128";
+  NSString *model = modelId.length > 0 ? modelId : @"eleven_multilingual_v2";
+  NSString *escapedVoiceId =
+      [voiceId stringByAddingPercentEncodingWithAllowedCharacters:
+                   [NSCharacterSet URLPathAllowedCharacterSet]];
+  NSString *escapedFormat =
+      [format stringByAddingPercentEncodingWithAllowedCharacters:
+                  [NSCharacterSet URLQueryAllowedCharacterSet]];
+  NSString *urlString =
+      [NSString stringWithFormat:@"https://api.elevenlabs.io/v1/text-to-speech/%@/stream?output_format=%@",
+                                 escapedVoiceId ?: @"",
+                                 escapedFormat ?: @""];
+  NSURL *url = [NSURL URLWithString:urlString];
+  if (!url) {
+    NSError *error = [NSError
+        errorWithDomain:@"AiVM"
+                   code:-1
+               userInfo:@{NSLocalizedDescriptionKey : @"invalid URL"}];
+    if (completion) {
+      completion(nil, error);
+    }
+    return;
+  }
+
+  NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
+  request.HTTPMethod = @"POST";
+  [request setValue:@"application/json" forHTTPHeaderField:@"Content-Type"];
+  [request setValue:@"audio/mpeg" forHTTPHeaderField:@"Accept"];
+  [request setValue:apiKey forHTTPHeaderField:@"xi-api-key"];
+
+  NSDictionary *body = @{
+    @"text" : text ?: @"",
+    @"model_id" : model ?: @""
+  };
+  NSError *jsonError = nil;
+  NSData *jsonData = [NSJSONSerialization dataWithJSONObject:body
+                                                     options:0
+                                                       error:&jsonError];
+  if (jsonError) {
+    if (completion) {
+      completion(nil, jsonError);
+    }
+    return;
+  }
+  request.HTTPBody = jsonData;
+
+  NSURLSessionConfiguration *config =
+      [NSURLSessionConfiguration defaultSessionConfiguration];
+  NSURLSession *session = [NSURLSession sessionWithConfiguration:config];
+  NSURLSessionDataTask *task =
+      [session dataTaskWithRequest:request
+                 completionHandler:^(NSData *_Nullable data,
+                                     NSURLResponse *_Nullable response,
+                                     NSError *_Nullable error) {
+                   if (error) {
+                     if (completion) {
+                       completion(nil, error);
+                     }
+                     return;
+                   }
+
+                   if (![response isKindOfClass:[NSHTTPURLResponse class]]) {
+                     NSError *respError = [NSError
+                         errorWithDomain:@"AiVM"
+                                    code:-1
+                                userInfo:@{
+                                  NSLocalizedDescriptionKey :
+                                      @"invalid response"
+                                }];
+                     if (completion) {
+                       completion(nil, respError);
+                     }
+                     return;
+                   }
+
+                   NSInteger status =
+                       ((NSHTTPURLResponse *)response).statusCode;
+                   if (status < 200 || status >= 300 || data.length == 0) {
+                     NSError *respError = [NSError
+                         errorWithDomain:@"AiVM"
+                                    code:status
+                                userInfo:@{
+                                  NSLocalizedDescriptionKey :
+                                      @"request failed"
+                                }];
+                     if (completion) {
+                       completion(nil, respError);
+                     }
+                     return;
+                   }
+
+                   if (completion) {
+                     completion(data, nil);
+                   }
+                 }];
+  [task resume];
+}
+
 @end