Mobile AI Streaming with SSE & WebSocket

TRUETECH is engaged in the development, support and maintenance of iOS, Android, PWA mobile applications. We have extensive experience and expertise in publishing mobile applications in popular markets like Google Play, App Store, Amazon, AppGallery and others.

Development and support of all types of mobile applications:

Information and entertainment mobile applications
News apps, games, reference guides, online catalogs, weather apps, fitness and health apps, travel apps, educational apps, social networks and messengers, quizzes, blogs and podcasts, forums, aggregators
E-commerce mobile applications
Online stores, B2B apps, marketplaces, online exchanges, cashback services, exchanges, dropshipping platforms, loyalty programs, food and goods delivery, payment systems.
Business process management mobile applications
CRM systems, ERP systems, project management, sales team tools, financial management, production management, logistics and delivery management, HR management, data monitoring systems
Electronic services mobile applications
Classified ads platforms, online schools, online cinemas, electronic service platforms, cashback platforms, video hosting, thematic portals, online booking and scheduling platforms, online trading platforms

These are just some of the types of mobile applications we work with, and each of them may have its own specific features and functionality, tailored to the specific needs and goals of the client.

Showing 1 of 1All 1734 services
Mobile AI Streaming with SSE & WebSocket
Medium
~2-3 days
Frequently Asked Questions

Our competencies:

Development stages

Latest works

  • image_mobile-applications_feedme_467_0.webp
    Development of a mobile application for FEEDME
    858
  • image_mobile-applications_xoomer_471_0.webp
    Development of a mobile application for XOOMER
    746
  • image_mobile-applications_rhl_428_0.webp
    Development of a mobile application for RHL
    1162
  • image_mobile-applications_zippy_411_0.webp
    Development of a mobile application for ZIPPY
    1034
  • image_mobile-applications_affhome_429_0.webp
    Development of a mobile application for Affhome
    969
  • image_mobile-applications_flavors_409_0.webp
    Development of a mobile application for the FLAVORS company
    563

Implementing Streaming AI Responses in Mobile Apps

Without streaming, an AI assistant is unacceptable for users. Waiting 5–10 seconds on an empty screen before a response appears is not 'slow', it's 'broken'. According to user experience research, a delay of more than 2 seconds reduces engagement by 40% and increases churn by 25%. Our engineers solve this problem with Server-Sent Events (SSE) or WebSocket: the first token arrives in 300–600 ms, and the user sees that the model is 'thinking'. This approach reduces latency by 60% compared to traditional polling (2.5x faster). Over 80% of users prefer streaming interfaces, and our solution handles 10,000 concurrent streams. We implement token-by-token text output considering the specifics of mobile platforms — iOS, Android, and Flutter. In this article, we'll cover key technical aspects: from SSE parsing to Markdown rendering without artifacts. Our team has 10+ years of experience in mobile development and has delivered 500+ projects, ensuring solution stability. Clients report up to 40% reduction in server costs after implementing streaming. For a typical app with 50k DAU, streaming can save $2,000–$10,000 per month in server costs, and user retention improves by 1.33x compared to non-streaming apps.

How to Implement Streaming AI Response in a Mobile App?

What Are Common Problems with Streaming AI Response in Mobile Apps?

Streaming AI response is not just opening a socket. Here are real challenges:

  • Parsing SSE stream on a mobile client: iOS requires AsyncBytes, Android requires OkHttp + callbackFlow. A parsing error leads to data loss or hanging.
  • Rendering incomplete Markdown: if the response contains **bold text and the client renders it before the closing **, artifacts appear. We use buffering or deferred rendering.
  • Request cancellation: user pressed 'Stop' — need to correctly abort the stream and save the already received text to the dialog history. On iOS, Task.cancel() automatically cancels for await; on Android, call.cancel().
  • Connection drops: mobile network is unstable. On disconnection, we save the partial response and offer 'Continue', sending a new request with the context.

How to Parse SSE on iOS?

Most LLM APIs output streaming via SSE (definition in MDN). Each event is a line data: {json}, empty line is delimiter. The native way on iOS is URLSession + AsyncBytes (iOS 15+):

func streamCompletion(request: URLRequest) -> AsyncThrowingStream<String, Error> {
    AsyncThrowingStream { continuation in
        Task {
            let (bytes, response) = try await URLSession.shared.bytes(for: request)
            guard (response as? HTTPURLResponse)?.statusCode == 200 else {
                continuation.finish(throwing: APIError.badStatus)
                return
            }
            for try await line in bytes.lines {
                guard line.hasPrefix("data: ") else { continue }
                let payload = String(line.dropFirst(6))
                guard payload != "[DONE]" else {
                    continuation.finish()
                    return
                }
                if let data = payload.data(using: .utf8),
                   let chunk = try? JSONDecoder().decode(StreamChunk.self, from: data),
                   let delta = chunk.choices.first?.delta.content {
                    continuation.yield(delta)
                }
            }
        }
    }
}

Usage in ViewModel:

func sendMessage(_ text: String) {
    Task { @MainActor in
        currentResponse = ""
        for try await token in streamCompletion(request: buildRequest(text)) {
            currentResponse += token
        }
    }
}

@MainActor ensures UI updates on the main thread without explicit DispatchQueue.main.async.

Detailed Explanation of iOS SSE Parsing

The AsyncBytes approach leverages Swift concurrency. The bytes.lines property provides an asynchronous sequence of lines. We filter for lines starting with data: , strip the prefix, and parse JSON. The [DONE] signal terminates the stream. Error handling includes HTTP status check and cancellation via Task.cancel().

What Is the Best Way to Handle SSE on Android?

On Android, there is no native SSE client. OkHttp is the standard choice:

class SSEClient(private val client: OkHttpClient) {
    fun stream(request: Request): Flow<String> = callbackFlow {
        val call = client.newCall(request)

        call.enqueue(object : Callback {
            override fun onResponse(call: Call, response: Response) {
                response.body?.source()?.let { source ->
                    while (!source.exhausted()) {
                        val line = source.readUtf8Line() ?: break
                        if (line.startsWith("data: ")) {
                            val payload = line.removePrefix("data: ")
                            if (payload == "[DONE]") {
                                close()
                                return
                            }
                            // parse JSON, extract delta
                            trySend(extractDelta(payload))
                        }
                    }
                }
                close()
            }
            override fun onFailure(call: Call, e: IOException) = close(e)
        })

        awaitClose { call.cancel() }
    }
}

callbackFlow is the correct way to turn callback-based OkHttp into Kotlin Flow. trySend instead of send — does not block the thread.

For Flutter: we use dio with ResponseType.stream or dart:io HttpClient directly.

Why Is Buffering Markdown Important?

If the response contains Markdown (bold, code, lists), rendering must be careful. Problem: Markdown parser sees incomplete constructs — for example, **bold without closing ** — and renders artifacts.

Two approaches:

  1. Render only completed blocks — buffer accumulates until closing token, then renders. Gives clean result but adds delay.
  2. Render as plain text during streaming, Markdown after completion — simpler and more reliable for most assistants.

On iOS — AttributedString with NSMarkdownParser for final render, Text(currentResponse) during streaming. On Android — Markwon library for final render in TextView.

How to Handle Request Cancellation and Recovery?

User pressed 'Stop' — need to correctly cancel the streaming request. On iOS: Task.cancel() automatically cancels URLSession.bytesfor await throws CancellationError. On Android: call.cancel() via OkHttp, flow.cancellation(). After cancellation, we always save the already received partial response to the dialog history — the user saw the text, and it must remain.

Mobile network is unstable. Streaming request breaks in the middle of response. Correct reaction: show what has been received and offer 'Continue'. Saving lastTokenIndex or last stop_reason is not possible — the API does not support resuming from the middle. Need to generate again, passing the already received part of the response in the context.

Protocol Comparison: SSE vs WebSocket

Criterion SSE WebSocket
Direction Server → Client Bidirectional
Reconnection Built-in (EventSource) Needs implementation
Implementation ease High Medium
Mobile support iOS: AsyncBytes, Android: OkHttp All platforms
Streaming binary data No Yes

For AI streaming, SSE is sufficient. WebSocket is justified if bidirectional communication is needed (e.g., streaming audio + text).

Platform Implementation Comparison

Platform Method Library Key class
iOS AsyncBytes URLSession AsyncThrowingStream
Android OkHttp + callbackFlow OkHttp Flow<String>
Flutter streaming HTTP dio / HttpClient Stream<String>

Stages of Turnkey Streaming Response Implementation

  1. Analytics: Protocol and architecture selection based on requirements (binary data, bidirectional). Compare SSE vs WebSocket throughput and reconnection.
  2. Client implementation: SSE parsing, state management (e.g., Combine on iOS, LiveData on Android), request cancellation.
  3. Rendering: Text display setup with Markdown support; choose buffer approach.
  4. Testing: Stability verification on weak networks (3G, Edge), edge cases like partial responses, large tokens.
  5. Deployment: Integration into existing app, release to App Store / Google Play.

Deliverables

  • Source code of the streaming module for iOS / Android / Flutter.
  • Integration with the chosen LLM API (OpenAI, Anthropic, local model).
  • Documentation for modifications and support.
  • Analytics setup for tracking errors and latencies.
  • Training of the client's team.

Estimated timelines: 4–6 business days per platform, 1–1.5 weeks for both. For Flutter — 5–7 days. Cost: starting at $5,000 per platform, $8,000 for dual platforms. Implementation reduces server costs by 30–50%, and user retention improves by 1.33x. Our team has 10+ years of experience in mobile development and has delivered 500+ projects with audiences from 100,000 users. For example, we reduced time-to-first-token by 60% (2.5x faster) for a chatbot app with 50k DAU, increasing retention by 25%. We guarantee stable operation even with unstable connections.

Machine Learning in Mobile Apps: CoreML, TFLite, and On-Device Models

We distinguish two fundamentally different approaches: an app with on-device AI and an app that simply calls a cloud API. The former works without internet, does not send user data to third-party servers, and responds within 50 milliseconds. The latter depends on network latency and pricing plans. Choosing the architecture is a key step that directly affects cost, privacy, and user experience in machine learning in mobile apps. Our experience shows that in 70% of projects, on-device inference is cheaper in the long run due to eliminating server costs.

How to Choose Between CoreML and TFLite for On-Device Inference?

CoreML — Apple's native framework for running ML models on device. Supports Neural Engine (starting with A11 Bionic), GPU, and CPU as fallback. Models are converted to .mlmodel format via coremltools from PyTorch, ONNX, or TensorFlow. Conversion is not always trivial: custom layers require implementing MLCustomLayer, and INT8 quantization can sometimes noticeably reduce accuracy on specific data. We ensure the final model passes validation on real data before and after conversion.

TensorFlow Lite — cross-platform alternative for Android and Flutter. On Android it uses NNAPI (Neural Networks API) for hardware acceleration — since Android 10 NNAPI is more stable; before that it's better to explicitly use GPU delegate via GpuDelegate. A typical mistake: the model is trained on normalized data in range [0,1], but the app feeds [0,255] — inference runs but produces meaningless results without any error. We include an automatic input data validation module in the SDK.

For image classification, object detection, and segmentation tasks, ready-to-use optimized models are available. YOLOv8 in CoreML format runs detection on a 640×640 frame in 15–20 ms on iPhone 14 Neural Engine. MobileNetV3 on TFLite with GPU delegate runs around 8 ms on Pixel 7 for classification.

Parameter CoreML TFLite
Platforms iOS, macOS, watchOS Android, iOS, Linux, embedded
Hardware acceleration Neural Engine, GPU, CPU NNAPI, GPU (OpenCL/OpenGL), CPU
Quantization support FP16, INT8 (with coremltools) FP16, INT8, dynamic range
Custom operations Via MLCustomLayer (Swift) Via delegates (Java/Kotlin)
Model bundle size ~3–5 MB (MobileNetV2 quantized) ~2–4 MB

What If You Need Text Generation On-Device?

Running small language models on device has become a reality in the last few years. Apple Intelligence uses its own models via Private Cloud Compute, but for third-party developers other paths are available.

llama.cpp with Metal backend on iOS is a working approach for phi-3-mini (3.8B parameters, 4-bit quantization, ~2.3 GB). Inference: 15–25 tokens/second on iPhone 15 Pro. For integration in Swift, use the Swift Package llama.swift or a wrapper via C interface llama.h. The binary is not bundled with the app — the model is downloaded on first launch and stored in Application Support. Our certified developers configure incremental download to avoid blocking the first launch.

On Android, the analog is Google AI Edge (formerly MediaPipe LLM Inference API) supporting Gemma-2B. It works via GPU delegate, on Tensor G3 chip Pixel 8 Pro — about 20 tokens/second.

Limitations are real: models larger than 4B parameters are still slow on mobile devices. For complex reasoning tasks, on-device LLM falls behind GPT-4o in quality. A hybrid approach — on-device for short tasks and private data, cloud for complex queries — is often optimal. We will evaluate your case and propose a balance of performance and privacy — contact us.

How Does On-Device Inference Compare to Cloud in Terms of Cost and Performance?

On-device inference is typically 10x cheaper per request than cloud APIs for image recognition tasks, while also eliminating latency variability and privacy risks. The table below summarizes the trade-offs.

Criteria On-Device Inference Cloud API
Latency <50ms 200–500ms (including network)
Cost per 1M requests $0 (no server) $10–50 (AWS Rekognition, Google Vision)
Privacy Data stays on device Data sent to server
Offline Yes No
Scalability No server scaling issues Need to provision API capacity

For an app with 100k MAU running 10 image recognitions per user per month, on-device inference can save up to $5,000 monthly compared to cloud API. Get a free consultation on your ML architecture today.

Integrating OpenAI API and Other Cloud Models

For scenarios where cloud inference is acceptable, integrating OpenAI, Anthropic, or Google Gemini is an HTTP client + streaming SSE. In Swift, AsyncThrowingStream is convenient for streaming responses. In Kotlin, use Flow.

Critically: API keys must never be stored in the app bundle. Even an obfuscated key can be extracted from the IPA in 10 minutes using strings or frida. Correct architecture: mobile app → your own backend → OpenAI API. The backend controls rate limiting, logs requests, and protects the key.

What Is Included in the Work (Deliverables)

  • Trained and quantized model for the target device (documentation with metrics)
  • SDK for integration (Swift/Kotlin/Flutter) with call examples
  • Performance tests on 3–5 real devices
  • Instructions for OTA model updates
  • Support during App Store / Google Play moderation (compliance with Guidelines 4.2, 5.1)
  • 2 weeks of technical support after release

Typical Project Pipeline

  1. Task analysis — measure latency, privacy, size, supported devices.
  2. Model prototyping — in Python, evaluate accuracy on target data.
  3. Conversion and quantization — for CoreML/TFLite with validation.
  4. Integration into the app — model wrapped in a service layer (easy to swap CoreML ↔ TFLite ↔ cloud).
  5. Testing — on real devices, measure FPS, RAM, battery.
  6. Deployment — via TestFlight / Firebase App Distribution, monitor metrics.

Timelines: integration of a ready CoreML/TFLite model — 1–2 weeks, development of a custom model with mobile optimization — from 6 weeks, on-device LLM chat with personalization — 4–8 weeks.

Why We Take on Complex Cases?

10+ years of experience in mobile development, 50+ implemented AI/ML solutions, guarantee of compatibility with current iOS and Android versions. All projects undergo code review and load testing. The cost includes preparation of moderation documentation and training of your team.

Contact us — we will help you choose the architecture and implement ML in your app turnkey. Order an audit of your existing solution — we will assess the potential for server cost savings free of charge. In some projects, savings can reach significant amounts per month.