Skip to content

Commit 8e2b15f

Browse files
committed
refactor(api): refactor CoUnitPromptGenerator to use a private method to create the Retrofit instance #51
This commit refactors the `CoUnitPromptGenerator` class to extract the creation of the `Retrofit` instance into a private method. This change improves code readability and maintainability by avoiding duplicate code and centralizing the creation logic.
1 parent 2e0de74 commit 8e2b15f

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

src/main/kotlin/cc/unitmesh/devti/counit/CoUnitPromptGenerator.kt

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,19 +13,14 @@ import retrofit2.converter.gson.GsonConverterFactory
1313

1414
@Service(Service.Level.PROJECT)
1515
class CoUnitPromptGenerator(val project: Project) {
16-
private var retrofit = Retrofit.Builder()
17-
.baseUrl(project.customRagSettings.serverAddress)
18-
.addConverterFactory(GsonConverterFactory.create())
19-
.build()
20-
2116
fun findIntention(input: String): String? {
22-
val service: CoUnitApi = retrofit.create(CoUnitApi::class.java)
17+
val service: CoUnitApi = coUnitApi()
2318
val body = service.explainQuery(input).execute().body()
2419
return body?.prompt
2520
}
2621

2722
fun semanticQuery(query: ExplainQuery): QueryResult {
28-
val service: CoUnitApi = retrofit.create(CoUnitApi::class.java)
23+
val service: CoUnitApi = coUnitApi()
2924
val englishQuery: QueryResponse? = service.query(query.query, PayloadType.OpenApi).execute().body()
3025
val hydeDoc: QueryResponse? = service.query(query.hypotheticalDocument, PayloadType.OpenApi).execute().body()
3126
val naturalLangQuery: QueryResponse? = service.query(query.natureLangQuery, PayloadType.OpenApi).execute().body()
@@ -36,4 +31,14 @@ class CoUnitPromptGenerator(val project: Project) {
3631
hydeDoc?.data ?: emptyList()
3732
)
3833
}
34+
35+
private fun coUnitApi(): CoUnitApi {
36+
val retrofit = Retrofit.Builder()
37+
.baseUrl(project.customRagSettings.serverAddress)
38+
.addConverterFactory(GsonConverterFactory.create())
39+
.build()
40+
41+
val service: CoUnitApi = retrofit.create(CoUnitApi::class.java)
42+
return service
43+
}
3944
}

0 commit comments

Comments
 (0)