Merge pull request #41 from elevenlabs/angelo/envs-support

Add environment parameter support

Author: AngeloGiacco

elevenlabs-sdk/src/main/java/io/elevenlabs/ConversationClientImpl.kt

@@ -42,7 +42,8 @@ internal object ConversationClientImpl {
             val tokenResponse = tokenService.fetchPublicAgentToken(
                 config.agentId!!,
                 config.overrides?.client?.source ?: "android_sdk",
-                config.overrides?.client?.version ?: BuildConfig.SDK_VERSION
+                config.overrides?.client?.version ?: BuildConfig.SDK_VERSION,
+                config.environment
             )
             config.copy(conversationToken = tokenResponse.token, agentId = null)
         } else {

elevenlabs-sdk/src/main/java/io/elevenlabs/ConversationConfig.kt

@@ -27,6 +27,7 @@ import io.elevenlabs.models.DisconnectionDetails
  * @param audioInputSampleRate Sample rate for audio recording in Hz (default: 48000 for high quality)
  * @param apiEndpoint Base URL for ElevenLabs API (default: "https://api.elevenlabs.io")
  * @param websocketUrl WebSocket URL for LiveKit WebRTC connection (default: "wss://livekit.rtc.elevenlabs.io")
+ * @param environment Optional environment name for the agent (defaults to "production" on the server)
  */
 data class ConversationConfig(
     val agentId: String? = null,
@@ -36,6 +37,7 @@ data class ConversationConfig(
     val audioInputSampleRate: Int = 48000,
     val apiEndpoint: String = "https://api.elevenlabs.io",
     val websocketUrl: String = "wss://livekit.rtc.elevenlabs.io",
+    val environment: String? = null,
     val overrides: Overrides? = null,
     val customLlmExtraBody: Map<String, Any>? = null,
     val dynamicVariables: Map<String, Any>? = null,

elevenlabs-sdk/src/main/java/io/elevenlabs/network/TokenService.kt

@@ -27,11 +27,12 @@ class TokenService(
      * @param agentId The ID of the public agent to get a token for
      * @param source Optional source identifier (defaults to "android_sdk")
      * @param version Optional version string (defaults to SDK version)
+     * @param environment Optional environment name (defaults to "production" on the server)
      * @return TokenResponse containing the token and connection details
      * @throws TokenServiceException if the request fails or returns an error
      */
-    suspend fun fetchPublicAgentToken(agentId: String, source: String, version: String): TokenResponse = withContext(Dispatchers.IO) {
-        val url = buildTokenUrl(agentId, source, version)
+    suspend fun fetchPublicAgentToken(agentId: String, source: String, version: String, environment: String? = null): TokenResponse = withContext(Dispatchers.IO) {
+        val url = buildTokenUrl(agentId, source, version, environment)
 
         val request = Request.Builder()
             .url(url)
@@ -69,10 +70,13 @@ class TokenService(
      * Build the URL for fetching conversation tokens
      *
      * @param agentId The agent ID to include in the request
+     * @param environment Optional environment name
      * @return Complete URL for the token request
      */
-    private fun buildTokenUrl(agentId: String, source: String, version: String): String {
-        return "$baseUrl/v1/convai/conversation/token?agent_id=$agentId&source=$source&version=$version"
+    private fun buildTokenUrl(agentId: String, source: String, version: String, environment: String? = null): String {
+        var url = "$baseUrl/v1/convai/conversation/token?agent_id=$agentId&source=$source&version=$version"
+        environment?.let { url += "&environment=$it" }
+        return url
     }
 }