spacy nlp ner

backend/internal/nlp/client.go

@@ -0,0 +1,80 @@
+package nlp
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"time"
+)
+
+type Client struct {
+	endpoint   string
+	httpClient *http.Client
+}
+
+func NewClient(endpoint string) *Client {
+	return &Client{
+		endpoint: endpoint,
+		httpClient: &http.Client{
+			Timeout: 60 * time.Second,
+		},
+	}
+}
+
+func (c *Client) Analyze(text string) (*AnalyzedFile, error) {
+	reqBody, err := json.Marshal(NLPRequest{Text: text})
+	if err != nil {
+		return nil, fmt.Errorf("marshal request: %w", err)
+	}
+
+	resp, err := c.httpClient.Post(
+		c.endpoint+"/nlp",
+		"application/json",
+		bytes.NewReader(reqBody),
+	)
+	if err != nil {
+		return nil, fmt.Errorf("call nlp server: %w", err)
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, fmt.Errorf("read response: %w", err)
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("nlp server returned %d: %s", resp.StatusCode, string(body))
+	}
+
+	var nlpResp NLPResponse
+	if err := json.Unmarshal(body, &nlpResp); err != nil {
+		return nil, fmt.Errorf("decode response: %w", err)
+	}
+
+	if nlpResp.ErrorCode != 0 {
+		return nil, fmt.Errorf("nlp error %d: %s", nlpResp.ErrorCode, nlpResp.Message)
+	}
+
+	result := &AnalyzedFile{
+		Entities:   nlpResp.Result.Entities,
+		NounChunks: nlpResp.Result.NounChunks,
+		Sentences:  nlpResp.Result.Sentences,
+	}
+
+	return result, nil
+}
+
+func (c *Client) HealthCheck() error {
+	resp, err := c.httpClient.Get(c.endpoint + "/health")
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("health check failed: status %d", resp.StatusCode)
+	}
+	return nil
+}