MCPcopy Create free account
hub / github.com/APIParkLab/APIPark / removeModel

Method removeModel

ai-provider/model-runtime/provider.go:311–340  ·  view source on GitHub ↗
(id string)

Source from the content-addressed store, hash-verified

309}
310
311func (r *ModelRegistry) removeModel(id string) {
312 r.models.Del(id)
313
314 // check node exist
315 node, exist := r.reverseMap[id]
316 if !exist {
317 return
318 }
319
320 // get lock
321 shard := r.getShard(node.typeKey)
322 shard.Lock()
323 defer shard.Unlock()
324
325 // delete node chain
326 if node.prev != nil {
327 node.prev.next = node.next
328 } else {
329 r.typeIndex[node.typeKey] = node.next
330 }
331 if node.next != nil {
332 node.next.prev = node.prev
333 }
334
335 // clean index
336 delete(r.reverseMap, id)
337 if r.typeIndex[node.typeKey] == nil {
338 delete(r.typeIndex, node.typeKey)
339 }
340}
341
342func (p *Provider) DefaultModel(modelType string) (IModel, bool) {
343 return p.registry.defaultModels.Get(modelType)

Callers 1

RemoveModelMethod · 0.80

Calls 2

getShardMethod · 0.95
DelMethod · 0.45

Tested by

no test coverage detected