thomas commited on
Commit ·
d5d78f3
1
Parent(s): 156199c
bugfix: fixed black issue
Browse files
Brain/src/rising_plugin/csv_embed.py
CHANGED
|
@@ -26,7 +26,7 @@ def csv_embed():
|
|
| 26 |
"""getting embed"""
|
| 27 |
|
| 28 |
|
| 29 |
-
def get_embed(data: str, setting:ReqModel) -> list[float]:
|
| 30 |
embeddings = OpenAIEmbeddings(openai_api_key=setting.openai_key)
|
| 31 |
return embeddings.embed_query(data)
|
| 32 |
|
|
|
|
| 26 |
"""getting embed"""
|
| 27 |
|
| 28 |
|
| 29 |
+
def get_embed(data: str, setting: ReqModel) -> list[float]:
|
| 30 |
embeddings = OpenAIEmbeddings(openai_api_key=setting.openai_key)
|
| 31 |
return embeddings.embed_query(data)
|
| 32 |
|