Skip to content

Commit

Permalink
update symmetry service properties on extension config changes
Browse files Browse the repository at this point in the history
  • Loading branch information
rjmacarthy committed Sep 4, 2024
1 parent 750696f commit 96560b2
Showing 1 changed file with 19 additions and 3 deletions.
22 changes: 19 additions & 3 deletions src/extension/symmetry-service.ts
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,10 @@ export class SymmetryService extends EventEmitter {
private _providerTopic: Buffer | undefined
private _emitterKey = ''
private _provider: SymmetryProvider | undefined
private _modelName = this._config.symmetryModelName
private _symmetryProvider = this._config.symmetryProvider
private _symmetryServerKey = this._config.symmetryServerKey


constructor(
view: WebviewView | undefined,
Expand All @@ -67,6 +71,11 @@ export class SymmetryService extends EventEmitter {
`${EVENT_NAME.twinnyGlobalContext}-${GLOBAL_STORAGE_KEY.autoConnectSymmetryProvider}`
)
if (autoConnectProvider) this.startSymmetryProvider()

workspace.onDidChangeConfiguration((event) => {
if (!event.affectsConfiguration('twinny')) return
this.updateConfig()
})
}

public connect = async (key: string) => {
Expand All @@ -80,7 +89,7 @@ export class SymmetryService extends EventEmitter {
this._serverPeer = peer
peer.write(
createSymmetryMessage(SYMMETRY_DATA_MESSAGE.requestProvider, {
modelName: this._config.symmetryModelName
modelName: this._modelName,
})
)
peer.on('data', (message: Buffer) => {
Expand Down Expand Up @@ -174,7 +183,7 @@ export class SymmetryService extends EventEmitter {

this.handleIncomingData(chunk, (response: StreamResponse) => {
const data = getChatDataFromProvider(
this._config.symmetryProvider,
this._symmetryProvider,
response
)
this._completion = this._completion + data
Expand Down Expand Up @@ -255,7 +264,7 @@ export class SymmetryService extends EventEmitter {
name: os.hostname(),
path: configPath,
public: true,
serverKey: this._config.symmetryServerKey,
serverKey: this._symmetryServerKey,
systemMessage: ''
})

Expand Down Expand Up @@ -306,4 +315,11 @@ export class SymmetryService extends EventEmitter {
public write(message: string) {
this._providerPeer?.write(message)
}

private updateConfig() {
this._config = workspace.getConfiguration('twinny')
this._symmetryProvider = this._config.symmetryProvider
this._symmetryServerKey = this._config.symmetryServerKey
this._modelName = this._config.symmetryModelName
}
}

0 comments on commit 96560b2

Please sign in to comment.