code cleanup

This commit is contained in:
Mark Wolters
2023-06-01 19:24:19 +00:00
parent 6100a43b80
commit 5551c1fb13
4 changed files with 37 additions and 3 deletions
@@ -88,9 +88,9 @@ public abstract class PineconeOpDispenser extends BaseOpDispenser<PineconeOp, Pi
}
protected Map<String, Value> generateMetadataMap(Map<String, Object> metadata_values_map) {
Map<String, Value> metadata_map = new HashMap<String,Value>();
Map<String, Value> metadata_map = new HashMap<>();
BiConsumer<String,Object> stringToValue = (key, val) -> {
Value targetval = null;
Value targetval;
if (val instanceof String) targetval = Value.newBuilder().setStringValue((String)val).build();
else if (val instanceof Number) targetval = Value.newBuilder().setNumberValue((((Number) val).doubleValue())).build();
else if (val instanceof List) targetval = Value.newBuilder().setListValue(generateListValue((List) val)).build();
@@ -0,0 +1,14 @@
scenarios:
default:
query: run driver=pinecone tags==blocks:query cycles=10
blocks:
query:
ops:
op1:
query: "test-index"
vector: "0.8602578079921012,0.12103044768221516,0.7737329191858439,0.4521093269320254,0.29351661477669416,0.4261807015226558,0.14131665592103335,0.882370813029422,0.4412833140430886,0.9916525700115515"
namespace: "example_namespace"
top_k: 1000
include_values: true
include_metadata: true
@@ -0,0 +1,20 @@
scenarios:
default:
upsert: run driver=pinecone tags==blocks:upsert cycles=1000 cycles===TEMPLATE(rampup-cycles,10000000) threads=auto
bindings:
id: Mod(<<keycount:1000000000>>); ToString() -> String
vector_value: ListSizedHashed(<<dimensions:5>>,HashRange(0.0f,100.0f) -> float); ToString() -> String
city: Cities()
blocks:
upsert:
ops:
op1:
upsert: "test-index"
namespace: "example_namespace"
upsert_vectors:
- id: {id}
values: {vector_value}
metadata:
city: {city}
@@ -1,6 +1,6 @@
scenarios:
default:
query: run driver=pinecone tags==blocks:upsert cycles=1000
upsert: run driver=pinecone tags==blocks:upsert cycles=1000
blocks:
upsert: