commonMain.tool.editor.TextEditor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of anthropic-sdk-kotlin-jvm Show documentation

Kotlin multiplatform client for accessing Ahtropic APIs

The newest version!

/*
 * Copyright 2024 Kazimierz Pogoda / Xemantic
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.xemantic.ai.anthropic.tool.editor

import com.xemantic.ai.anthropic.cache.CacheControl
import com.xemantic.ai.anthropic.tool.BuiltInTool
import com.xemantic.ai.anthropic.tool.ToolInput
import kotlinx.serialization.ExperimentalSerializationApi
import kotlinx.serialization.SerialName
import kotlinx.serialization.Serializable
import kotlinx.serialization.Transient

@Serializable
@SerialName("str_replace_editor")
@OptIn(ExperimentalSerializationApi::class)
data class TextEditor(
  override val cacheControl: CacheControl? = null
) : BuiltInTool(
  name = "str_replace_editor",
  type = "text_editor_20241022"
) {

  init {
    inputSerializer = Input.serializer()
  }

  @Serializable
  data class Input(
    val command: Command,
    @SerialName("file_text")
    val fileText: String? = null,
    @SerialName("insert_line")
    val insertLine: Int? = null,
    @SerialName("new_str")
    val newStr: String? = null,
    @SerialName("old_str")
    val oldStr: String? = null,
    @SerialName("path")
    val path: String,
    @SerialName("view_range")
    val viewRange: Int? = 0
  ) : ToolInput() {

    @Transient
    lateinit var service: TextEditorService

    init {
      use {
        when (command) {
          Command.VIEW -> service.view(path)
          else -> TODO("not implemented yet")
        }
      }
    }

  }

}

@Serializable
enum class Command {
  @SerialName("view")
  VIEW,
  @SerialName("create")
  CREATE,
  @SerialName("str_replace")
  STR_REPLACE,
  @SerialName("insert")
  INSERT,
  @SerialName("undo_edit")
  UNDO_EDIT
}

interface TextEditorService {

  fun view(path: String)

}