solana-agent 0.0.21__tar.gz → 0.0.22__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {solana_agent-0.0.21 → solana_agent-0.0.22}/PKG-INFO +1 -1
- {solana_agent-0.0.21 → solana_agent-0.0.22}/pyproject.toml +1 -1
- {solana_agent-0.0.21 → solana_agent-0.0.22}/solana_agent/ai.py +3 -3
- {solana_agent-0.0.21 → solana_agent-0.0.22}/LICENSE +0 -0
- {solana_agent-0.0.21 → solana_agent-0.0.22}/README.md +0 -0
- {solana_agent-0.0.21 → solana_agent-0.0.22}/solana_agent/__init__.py +0 -0
|
@@ -141,7 +141,7 @@ class AI:
|
|
|
141
141
|
Pinecone(api_key=pinecone_api_key) if pinecone_api_key else None
|
|
142
142
|
)
|
|
143
143
|
self._pinecone_index_name = pinecone_index_name if pinecone_index_name else None
|
|
144
|
-
self.
|
|
144
|
+
self.kb = (
|
|
145
145
|
self._pinecone.Index(
|
|
146
146
|
self._pinecone_index_name) if self._pinecone else None
|
|
147
147
|
)
|
|
@@ -261,7 +261,7 @@ class AI:
|
|
|
261
261
|
input=query,
|
|
262
262
|
model=self._openai_embedding_model,
|
|
263
263
|
)
|
|
264
|
-
search_results = self.
|
|
264
|
+
search_results = self.kb.query(
|
|
265
265
|
vector=response.data[0].embedding,
|
|
266
266
|
top_k=limit,
|
|
267
267
|
include_metadata=True,
|
|
@@ -302,7 +302,7 @@ class AI:
|
|
|
302
302
|
input=values,
|
|
303
303
|
model=self._openai_embedding_model,
|
|
304
304
|
)
|
|
305
|
-
self.
|
|
305
|
+
self.kb.upsert(
|
|
306
306
|
vectors=[
|
|
307
307
|
{
|
|
308
308
|
"id": uuid.uuid4().hex,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|