File tree Expand file tree Collapse file tree
BingusApi/EmbeddingServices Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -20,17 +20,17 @@ public RocksDbStore(RocksDb rocksDb)
2020
2121 private static string DeserializeString ( byte [ ] value ) => Encoding . UTF8 . GetString ( value ) ;
2222
23- public void Add ( string key , Vector < float > embedding )
23+ public void Put ( string key , Vector < float > embedding )
2424 {
25- Add ( key , RocksDbSerializer . SerializeVector ( embedding ) ) ;
25+ Put ( key , RocksDbSerializer . SerializeVector ( embedding ) ) ;
2626 }
2727
28- public void Add ( string key , float [ ] embedding )
28+ public void Put ( string key , float [ ] embedding )
2929 {
30- Add ( key , RocksDbSerializer . SerializeArray ( embedding ) ) ;
30+ Put ( key , RocksDbSerializer . SerializeArray ( embedding ) ) ;
3131 }
3232
33- public void Add ( string key , byte [ ] data )
33+ public void Put ( string key , byte [ ] data )
3434 {
3535 rocksDb . Put ( SerializeString ( key ) , data ) ;
3636 }
@@ -107,7 +107,7 @@ public static RocksDbStore Create(
107107 modelUid
108108 ) ;
109109 }
110- store . Add ( ModelUidKey , SerializeString ( modelUid ) ) ;
110+ store . Put ( ModelUidKey , SerializeString ( modelUid ) ) ;
111111 }
112112
113113 return store ;
Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ public void AddItems(
5353 if ( embedding == null )
5454 {
5555 embedding = _sentenceEncoder . ComputeEmbeddingVector ( query ) ;
56- _embeddingStore ? . Add ( query , embedding ) ;
56+ _embeddingStore ? . Put ( query , embedding ) ;
5757 }
5858
5959 var faqEntry = new FaqEntry ( )
@@ -78,7 +78,7 @@ int numResults
7878 _embeddingCache ? . GetRaw ( query )
7979 ?? _embeddingStore ? . GetRaw ( query )
8080 ?? _sentenceEncoder . ComputeEmbedding ( query ) ;
81- _embeddingCache ? . Add ( query , vector ) ;
81+ _embeddingCache ? . Put ( query , vector ) ;
8282 return _hnswHandler . SearchItems ( vector , numResults ) ;
8383 }
8484 }
Original file line number Diff line number Diff line change @@ -4,8 +4,8 @@ namespace BingusLib.FaqHandling
44{
55 public interface IEmbeddingHandler
66 {
7- void Add ( string key , Vector < float > embedding ) ;
8- void Add ( string key , float [ ] embedding ) ;
7+ void Put ( string key , Vector < float > embedding ) ;
8+ void Put ( string key , float [ ] embedding ) ;
99 Vector < float > ? Get ( string key ) ;
1010 float [ ] ? GetRaw ( string key ) ;
1111 bool Has ( string key ) ;
You can’t perform that action at this time.
0 commit comments