github-mirror/Telegram-iOS
1
0
Fork 0
mirror of https://github.com/TelegramMessenger/Telegram-iOS.git synced 2026-07-06 03:33:41 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

composeMessageWithAI

Browse source
This commit is contained in:
Mikhail Filimonov 2026-03-17 08:25:16 +01:00
parent b64d4a23fe
commit 25d601920e
2 changed files with 41 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
submodules/TelegramCore/Sources/TelegramEngine/Messages/TelegramEngineMessages.swift
View file

@ -635,6 +635,10 @@ public extension TelegramEngine {
}
}
public func composeMessageWithAI(text: String, entities: [MessageTextEntity], proofread: Bool = false, translateToLang: String? = nil, changeTone: String? = nil, emojify: Bool = false) -> Signal<(String, [MessageTextEntity]), TranslationError> {
return _internal_composeMessageWithAI(network: self.account.network, text: text, entities: entities, proofread: proofread, translateToLang: translateToLang, changeTone: changeTone, emojify: emojify)
}
public func translate(texts: [(String, [MessageTextEntity])], toLang: String, tone: TranslationTone = .neutral) -> Signal<[(String, [MessageTextEntity])], TranslationError> {
return _internal_translateTexts(network: self.account.network, texts: texts, toLang: toLang, tone: tone)
}

37
submodules/TelegramCore/Sources/TelegramEngine/Messages/Translate.swift
View file

@ -68,6 +68,43 @@ func _internal_translate(network: Network, text: String, toLang: String, entitie
}
}
func _internal_composeMessageWithAI(network: Network, text: String, entities: [MessageTextEntity], proofread: Bool = false, translateToLang: String? = nil, changeTone: String? = nil, emojify: Bool = false) -> Signal<(String, [MessageTextEntity]), TranslationError> {
var flags: Int32 = 0
if proofread {
flags |= (1 << 0)
}
if translateToLang != nil {
flags |= (1 << 1)
}
if changeTone != nil {
flags |= (1 << 2)
}
if emojify {
flags |= (1 << 3)
}
let apiText: Api.TextWithEntities = .textWithEntities(.init(text: text, entities: apiEntitiesFromMessageTextEntities(entities, associatedPeers: SimpleDictionary())))
return network.request(Api.functions.messages.composeMessageWithAI(flags: flags, text: apiText, translateToLang: translateToLang, changeTone: changeTone))
|> mapError { error -> TranslationError in
if error.errorDescription.hasPrefix("FLOOD_WAIT") {
return .limitExceeded
} else if error.errorDescription == "INPUT_TEXT_EMPTY" {
return .textIsEmpty
} else if error.errorDescription == "INPUT_TEXT_TOO_LONG" {
return .textTooLong
} else {
return .generic
}
}
|> map { result -> (String, [MessageTextEntity]) in
switch result {
case let .textWithEntities(data):
return (data.text, messageTextEntitiesFromApiEntities(data.entities))
}
}
}
func _internal_translateTexts(network: Network, texts: [(String, [MessageTextEntity])], toLang: String, tone: TranslationTone = .neutral) -> Signal<[(String, [MessageTextEntity])], TranslationError> {
var flags: Int32 = 0
flags |= (1 << 1)