Spaces:
Sleeping
Sleeping
Commit ·
ca748a4
1
Parent(s): 7c377b6
updated aggent
Browse files
agent.py
CHANGED
|
@@ -25,7 +25,8 @@ class CodingAgent:
|
|
| 25 |
self.conn.execute("""CREATE TABLE IF NOT EXISTS memory (id INTEGER PRIMARY KEY, query TEXT, response TEXT)""")
|
| 26 |
|
| 27 |
def embed_chunks(self, texts):
|
| 28 |
-
return self.embedder.encode(texts)
|
|
|
|
| 29 |
|
| 30 |
def ingest_file(self, filepath):
|
| 31 |
chunks = []
|
|
|
|
| 25 |
self.conn.execute("""CREATE TABLE IF NOT EXISTS memory (id INTEGER PRIMARY KEY, query TEXT, response TEXT)""")
|
| 26 |
|
| 27 |
def embed_chunks(self, texts):
|
| 28 |
+
return self.embedder.encode(texts, convert_to_numpy=True)
|
| 29 |
+
|
| 30 |
|
| 31 |
def ingest_file(self, filepath):
|
| 32 |
chunks = []
|