starred/minne
1
0
Fork 0
mirror of https://github.com/perstarkse/minne.git synced 2026-04-21 16:31:23 +02:00
Code Issues Packages Projects Releases 10 Wiki Activity

feat:completed ingress

Browse Source
This commit is contained in:
Per Stark
2024-11-20 16:09:24 +01:00
parent 4d6510ee2e
commit ec347d524c
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
rustfmt.toml Normal file
View File

@@ -0,0 +1 @@
edition = "2021"

11
src/models/text_content.rs
View File

@@ -5,7 +5,7 @@ use crate::{
}; };
use async_openai::error::OpenAIError; use async_openai::error::OpenAIError;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use surrealdb::{engine::remote::ws::Client, Surreal}; use surrealdb::{engine::remote::ws::Client, sql::Thing, Surreal};
use text_splitter::TextSplitter; use text_splitter::TextSplitter;
use thiserror::Error; use thiserror::Error;
use tracing::{debug, info}; use tracing::{debug, info};
@@ -83,7 +83,7 @@ async fn get_related_nodes(
id: String, id: String,
db_client: &Surreal<Client>, db_client: &Surreal<Client>,
) -> Result<Vec<KnowledgeEntity>, ProcessingError> { ) -> Result<Vec<KnowledgeEntity>, ProcessingError> {
let query = format!("SELECT -> knowledge_relationship -> knowledge_entity as related_nodes FROM knowledge_entity WHERE source_id = `{}`", id); let query = format!("SELECT * FROM knowledge_entity WHERE source_id = '{}'", id);
// let query = format!("SELECT * FROM knowledge_entity WHERE in OR out {}", id); // let query = format!("SELECT * FROM knowledge_entity WHERE in OR out {}", id);
let related_nodes: Vec<KnowledgeEntity> = db_client.query(query).await?.take(0)?; let related_nodes: Vec<KnowledgeEntity> = db_client.query(query).await?.take(0)?;
@@ -101,7 +101,7 @@ impl TextContent {
self.store_text_content(&db_client).await?; self.store_text_content(&db_client).await?;
let closest_text_content: Vec<TextChunk> = vector_comparison( let closest_text_content: Vec<TextChunk> = vector_comparison(
4, 3,
self.text.clone(), self.text.clone(),
&db_client, &db_client,
"text_chunk".to_string(), "text_chunk".to_string(),
@@ -110,9 +110,10 @@ impl TextContent {
.await?; .await?;
for node in closest_text_content { for node in closest_text_content {
info!("{}-{}", node.id, node.source_id);
let related_nodes = get_related_nodes(node.source_id, &db_client).await?; let related_nodes = get_related_nodes(node.source_id, &db_client).await?;
info!("{:?}", related_nodes); for related_node in related_nodes {
info!("{:?}", related_node.name);
}
} }
panic!("STOPPING"); panic!("STOPPING");

2
src/routes/file.rs
View File

@@ -1,5 +1,4 @@
use std::sync::Arc; use std::sync::Arc;
use axum::{ use axum::{
extract::Path, response::IntoResponse, Extension, Json extract::Path, response::IntoResponse, Extension, Json
}; };
@@ -8,7 +7,6 @@ use serde_json::json;
use tempfile::NamedTempFile; use tempfile::NamedTempFile;
use tracing::info; use tracing::info;
use uuid::Uuid; use uuid::Uuid;
use crate::{ use crate::{
models::file_info::{FileError, FileInfo}, models::file_info::{FileError, FileInfo},
surrealdb::SurrealDbClient, surrealdb::SurrealDbClient,